Skip to content

Commit 4be43af

Browse files
author
Marco Gorelli
committed
lint
1 parent f5d9088 commit 4be43af

File tree

3 files changed

+18
-14
lines changed

3 files changed

+18
-14
lines changed

pandas/core/reshape/reshape.py

+7-8
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from functools import partial
22
import itertools
3-
from typing import List, Optional, Union
3+
from typing import List, Optional, Set, Union
44

55
import numpy as np
66

@@ -751,9 +751,7 @@ def _convert_level_number(level_num, columns):
751751
return result
752752

753753

754-
def from_dummies(
755-
data, prefix=None, prefix_sep="_", dtype="category"
756-
) -> "DataFrame":
754+
def from_dummies(data, prefix=None, prefix_sep="_", dtype="category") -> "DataFrame":
757755
"""
758756
The inverse transformation of ``pandas.get_dummies``.
759757
@@ -833,10 +831,10 @@ def from_dummies(
833831
"If no columns contain `prefix_sep`, you must"
834832
" pass a value to `prefix` with which to name"
835833
" the decoded columns."
836-
)
834+
)
837835
# If no column contains `prefix_sep`, we add `prefix`_`prefix_sep` to
838836
# each column.
839-
out = data.rename(columns = lambda x: f'{prefix}{prefix_sep}{x}').copy()
837+
out = data.rename(columns=lambda x: f"{prefix}{prefix_sep}{x}").copy()
840838
columns_to_decode = out.columns
841839
else:
842840
out = data.copy()
@@ -846,7 +844,7 @@ def from_dummies(
846844
if prefix is None:
847845
# If no prefix has been passed, extract it from columns containing
848846
# `prefix_sep`
849-
seen = set()
847+
seen: Set[str] = set()
850848
prefix = []
851849
for i in columns_to_decode:
852850
i = i.split(prefix_sep)[0]
@@ -879,7 +877,8 @@ def _validate_values(data):
879877
_validate_values(data_to_decode[cols])
880878
out = out.drop(cols, axis=1)
881879
out[prefix_] = Series(
882-
np.array(labels)[np.argmax(data_to_decode[cols].to_numpy(), axis=1)], dtype=dtype
880+
np.array(labels)[np.argmax(data_to_decode[cols].to_numpy(), axis=1)],
881+
dtype=dtype,
883882
)
884883
return out
885884

pandas/tests/mytest.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
import pandas as pd
22

3+
34
def test_me():
45
pd.eval(
56
"""
67
A = df.A - df.B
78
B = df.A + df.B
89
""",
910
target=pd.DataFrame(),
10-
)
11+
)

pandas/tests/reshape/test_from_dummies.py

+9-5
Original file line numberDiff line numberDiff line change
@@ -45,14 +45,18 @@ def test_prefix_sep(prefix_sep, input_dict):
4545
expected = pd.DataFrame({"col1": ["a", "a", "b"]}, dtype="category")
4646
tm.assert_frame_equal(result, expected)
4747

48+
4849
def test_no_prefix():
4950
df = pd.DataFrame({"a": [1, 1, 0], "b": [0, 0, 1]})
50-
result = pd.from_dummies(df, prefix='letter')
51-
expected = pd.DataFrame({'letter': ['a', 'a', 'b']}, dtype='category')
51+
result = pd.from_dummies(df, prefix="letter")
52+
expected = pd.DataFrame({"letter": ["a", "a", "b"]}, dtype="category")
5253
tm.assert_frame_equal(result, expected)
5354

55+
5456
def test_multiple_columns():
55-
df = pd.DataFrame({"col1_a": [1, 0], "col1_b": [0, 1], "col2_a": [0, 0], "col2_c": [1, 1]})
57+
df = pd.DataFrame(
58+
{"col1_a": [1, 0], "col1_b": [0, 1], "col2_a": [0, 0], "col2_c": [1, 1]}
59+
)
5660
result = pd.from_dummies(df)
57-
expected = pd.DataFrame({'col1': ['a', 'b'], 'col2': ['c', 'c']}, dtype='category')
58-
tm.assert_frame_equal(result, expected)
61+
expected = pd.DataFrame({"col1": ["a", "b"], "col2": ["c", "c"]}, dtype="category")
62+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)