python - 如何将条件聚合应用于系列？

Question

对单个列的 groupby 应用聚合是:

df.groupby('column_name1')['column_name2'].agg(sum)

我有数据，如果 column1 包含值 A，我想在 column2 上应用 sum()，如果值为 B，我想应用 count()。

完成此操作的语法是什么？

我试图用 if 语句编写一个 for 循环，但我希望有更好的方法来完成它。

Answer 1

尝试使用下面两行:

df[df['column_name1'].str.contains('A')]=df[df['column_name1'].str.contains('A')].groupby('column_name1')['column_name2'].agg(sum)
df[df['column_name1'].str.contains('B')]=df[df['column_name1'].str.contains('B')].groupby('column_name1')['column_name2'].agg('count')