Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ENH] generic select function #1187

Merged
merged 13 commits into from
Nov 8, 2022
Merged

[ENH] generic select function #1187

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
e311ce6
fix blank note output
samukweku Nov 1, 2022
47ad9d8
Merge branch 'dev' of https://github.com/pyjanitor-devs/pyjanitor int…
samukweku Nov 2, 2022
f6a6e6c
return whatever user passes, even if they r duplicates
samukweku Nov 2, 2022
0325ada
add DropLabel for dropping columns
samukweku Nov 2, 2022
9c56c15
generic select with tests
samukweku Nov 3, 2022
b28082a
update test_select.py
samukweku Nov 3, 2022
f958da5
update changelog
samukweku Nov 3, 2022
735ad21
update docs in select
samukweku Nov 4, 2022
00ff25d
add version notifications
samukweku Nov 5, 2022
035a5c1
update docs for DropLabel class
samukweku Nov 5, 2022
c78444d
update admonition
samukweku Nov 6, 2022
d80baf3
Merge branch 'dev' of https://github.com/pyjanitor-devs/pyjanitor int…
samukweku Nov 7, 2022
78620cb
ensure booleans are converted into arrays
samukweku Nov 7, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
ensure booleans are converted into arrays
  • Loading branch information
@samukweku
samukweku committed Nov 7, 2022
commit 78620cb5c90c5911dd84dbac1c64ece564cb5e91
11 changes: 6 additions & 5 deletions janitor/functions/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -476,7 +476,7 @@ def _index_dispatch(arg, df, axis): # noqa: F811
f"{arg} is a boolean dtype and has wrong length: "
f"{len(arg)} instead of {len(index)}"
)
return arg
return np.asanyarray(arg)
try:

if isinstance(arg, pd.Series):
Expand Down Expand Up @@ -540,11 +540,12 @@ def _index_dispatch(arg, df, axis): # noqa: F811
# or materialized if possible;
# this offers more performance
if len(indices) == 1:
if isinstance(indices[0], int):
if is_scalar(indices[0]):
return indices
if is_list_like(indices[0]):
return np.asanyarray(indices[0])
return indices[0]
indices = indices[0]
if is_list_like(indices):
indices = np.asanyarray(indices)
return indices
contents = []
for arr in indices:
if is_list_like(arr):
Expand Down