Pandas 在满足条件的情况下计算列内的出现次数

Question

我正在尝试通过分组来计数。请参阅下面的输入和输出。

输入：

df = pd.DataFrame()

df['col1'] = ['a','a','a','a','b','b','b']
df['col2'] = [4,4,5,5,6,7,8]
df['col3'] = [1,1,1,1,1,1,1]

输出：

   col4
0     2
1     2
2     2
3     2
4     1
5     1
6     1

尝试通过以下方式玩弄 groupby 和 count：

s = df.groupby(['col1','col2'])['col3'].sum()

我得到的输出是

a     4       2
      5       2
b     6       1
      7       1
      8       1

如何将它添加为主 df 上的列。

谢谢你！

Answer 1

使用transformlen或size：

df['count'] = df.groupby(['col1','col2'])['col3'].transform(len)
print (df)
  col1  col2  col3  count
0    a     4     1      2
1    a     4     1      2
2    a     5     1      2
3    a     5     1      2
4    b     6     1      1
5    b     7     1      1
6    b     8     1      1

---

df['count'] = df.groupby(['col1','col2'])['col3'].transform('size')
print (df)
  col1  col2  col3  count
0    a     4     1      2
1    a     4     1      2
2    a     5     1      2
3    a     5     1      2
4    b     6     1      1
5    b     7     1      1
6    b     8     1      1

但col3列不是必须的，你可以使用col1或col2：

df = pd.DataFrame()

df['col1'] = ['a','a','a','a','b','b','b']
df['col2'] = [4,4,5,5,6,7,8]

df['count'] = df.groupby(['col1','col2'])['col1'].transform(len)
df['count1'] = df.groupby(['col1','col2'])['col2'].transform(len)
print (df)
  col1  col2 count  count1
0    a     4     2       2
1    a     4     2       2
2    a     5     2       2
3    a     5     2       2
4    b     6     1       1
5    b     7     1       1
6    b     8     1       1

Answer 2

试试这个，

df['count'] = df.groupby(['col1','col2'])['col3'].transform(sum)
print (df)
  col1  col2  col3  count
0    a     4     1    2
1    a     4     1    2
2    a     5     1    2
3    a     5     1    2
4    b     6     1    1
5    b     7     1    1
6    b     8     1    1