Skip to content

TST: groupby apply with indexing and column aggregation returns the column #7002 #34647

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

Merged
merged 12 commits into from
Jun 14, 2020
Merged

TST: groupby apply with indexing and column aggregation returns the column #7002 #34647

Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
2e77cef
TST: groupby apply with indexing and colunm aggregation returns the c…
fangchenli Jun 8, 2020
99fee1f
TST: groupby apply with indexing and column aggregation returns the c…
fangchenli Jun 8, 2020
e3a49fe
TST: groupby apply with indexing and column aggregation returns the c…
fangchenli Jun 8, 2020
eef7d59
TST/CLN: reformat (#7002)
fangchenli Jun 8, 2020
cfdbcf0
TST: indent (#7002)
fangchenli Jun 8, 2020
e8f181c
TST: indent (#7002)
fangchenli Jun 8, 2020
0bd567c
TST: deterministic test case (#7002)
fangchenli Jun 8, 2020
787d9ba
TST: explicitly constructed expected DataFrame (#7002)
fangchenli Jun 9, 2020
8139d8e
TST: explicitly constructed expected DataFrame (#7002)
fangchenli Jun 9, 2020
c8d5fca
TST: explicitly constructed expected DataFrame (#7002)
fangchenli Jun 9, 2020
e25128e
Merge remote-tracking branch 'upstream/master' into test-groupby-appl…
fangchenli Jun 14, 2020
5936715
TST: revert unrelated changes
fangchenli Jun 14, 2020
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
13 changes: 13 additions & 0 deletions pandas/tests/groupby/test_apply.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -961,3 +961,16 @@ def fn(x):
name="col2",
)
tm.assert_series_equal(result, expected)


def test_apply_function_with_indexing_return_column():
# GH: 7002
df = DataFrame(
{
"foo1": ["one", "two", "two", "three", "one", "two"],
"foo2": [1, 2, 4, 4, 5, 6],
}
)
result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())
expected = DataFrame({"foo1": ["one", "three", "two"], "foo2": [3.0, 4.0, 4.0]})
tm.assert_frame_equal(result, expected)