Skip to content

Commit 8db200c

Browse files
fixed pre-commit checks
1 parent 2000f18 commit 8db200c

File tree

3 files changed

+20
-16
lines changed

3 files changed

+20
-16
lines changed

pandas/core/strings/accessor.py

Lines changed: 19 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2422,23 +2422,21 @@ def get_dummies(
24222422
)
24232423
from pandas.core.reshape.encoding import get_dummies
24242424

2425-
input_series = Series(self._data) if isinstance(self._data, ABCIndex) else self._data
2425+
input_series = (
2426+
Series(self._data) if isinstance(self._data, ABCIndex) else self._data
2427+
)
24262428
string_series = input_series.apply(lambda x: str(x) if not isna(x) else x)
24272429
split_series = string_series.str.split(sep, expand=True).stack()
24282430
valid_split_series = split_series[
2429-
(split_series.astype(str) != 'None') &
2430-
~(split_series.index.get_level_values(0).duplicated(keep='first') & split_series.isna())
2431+
(split_series.astype(str) != "None")
2432+
& ~(
2433+
split_series.index.get_level_values(0).duplicated(keep="first")
2434+
& split_series.isna()
2435+
)
24312436
]
24322437

24332438
dummy_df = get_dummies(
2434-
valid_split_series,
2435-
None,
2436-
None,
2437-
dummy_na,
2438-
None,
2439-
sparse,
2440-
False,
2441-
dtype
2439+
valid_split_series, None, None, dummy_na, None, sparse, False, dtype
24422440
)
24432441
grouped_dummies = dummy_df.groupby(level=0)
24442442
if dtype == bool:
@@ -2447,7 +2445,9 @@ def get_dummies(
24472445
result_df = grouped_dummies.sum()
24482446

24492447
if isinstance(prefix, str):
2450-
result_df.columns = [f"{prefix}{prefix_sep}{col}" for col in result_df.columns]
2448+
result_df.columns = [
2449+
f"{prefix}{prefix_sep}{col}" for col in result_df.columns
2450+
]
24512451
elif isinstance(prefix, dict):
24522452
if len(prefix) != len(result_df.columns):
24532453
len_msg = (
@@ -2456,7 +2456,9 @@ def get_dummies(
24562456
f"({len(result_df.columns)})."
24572457
)
24582458
raise ValueError(len_msg)
2459-
result_df.columns = [f"{prefix[col]}{prefix_sep}{col}" for col in result_df.columns]
2459+
result_df.columns = [
2460+
f"{prefix[col]}{prefix_sep}{col}" for col in result_df.columns
2461+
]
24602462
elif isinstance(prefix, list):
24612463
if len(prefix) != len(result_df.columns):
24622464
len_msg = (
@@ -2465,7 +2467,10 @@ def get_dummies(
24652467
f"({len(result_df.columns)})."
24662468
)
24672469
raise ValueError(len_msg)
2468-
result_df.columns = [f"{prefix[i]}{prefix_sep}{col}" for i, col in enumerate(result_df.columns)]
2470+
result_df.columns = [
2471+
f"{prefix[i]}{prefix_sep}{col}"
2472+
for i, col in enumerate(result_df.columns)
2473+
]
24692474

24702475
if isinstance(self._data, ABCIndex):
24712476
return MultiIndex.from_frame(result_df)

pandas/core/strings/object_array.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from __future__ import annotations
22

3-
import functools
43
import re
54
import textwrap
65
from typing import (

pandas/tests/strings/test_get_dummies.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def test_get_dummies_with_name_dummy_index():
5353
)
5454
tm.assert_index_equal(result, expected)
5555

56+
5657
def test_get_dummies_with_prefix(any_string_dtype):
5758
s = Series(["a|b", "a|c", np.nan], dtype=any_string_dtype)
5859
result = s.str.get_dummies(sep="|", prefix="prefix")
@@ -119,4 +120,3 @@ def test_get_dummies_with_prefix_dict(any_string_dtype):
119120
columns=["alpha_a", "beta_b", "gamma_c"],
120121
)
121122
tm.assert_frame_equal(result, expected)
122-

0 commit comments

Comments
 (0)