Skip to content

Commit 777482f

Browse files
committed
first value_counts commit
1 parent 626b651 commit 777482f

File tree

2 files changed

+27
-1
lines changed

2 files changed

+27
-1
lines changed

pandas/core/frame.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6922,7 +6922,7 @@ def value_counts(
69226922
69236923
Parameters
69246924
----------
6925-
subset : list-like, optional
6925+
subset : mapping, function, label, list of labels, optional
69266926
Columns to use when counting unique combinations.
69276927
normalize : bool, default False
69286928
Return proportions rather than frequencies.
@@ -7014,6 +7014,13 @@ def value_counts(
70147014
John Smith 1
70157015
NaN 1
70167016
dtype: int64
7017+
7018+
>>> df.value_counts("first_name")
7019+
first_name
7020+
John 2
7021+
Anne 1
7022+
Beth 1
7023+
dtype: int64
70177024
"""
70187025
if subset is None:
70197026
subset = self.columns.tolist()

pandas/tests/frame/methods/test_value_counts.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,3 +144,22 @@ def test_data_frame_value_counts_dropna_false(nulls_fixture):
144144
)
145145

146146
tm.assert_series_equal(result, expected)
147+
148+
149+
def test_data_frame_value_counts_subset(nulls_fixture):
150+
# GH 50829
151+
df = pd.DataFrame(
152+
{
153+
"first_name": ["John", "Anne", "John", "Beth"],
154+
"middle_name": ["Smith", nulls_fixture, nulls_fixture, "Louise"],
155+
},
156+
)
157+
result = df.value_counts("first_name")
158+
expected = pd.Series(
159+
data=[2, 1, 1],
160+
index=pd.MultiIndex.from_arrays(
161+
[["John", "Anne", "Beth"]], names=["first_name"]
162+
),
163+
)
164+
165+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)