总结熊猫数据框中所有可能的组合

Question

我有一个包含几列的 df，但其中两列看起来像：

number output colour
1         1    green
2         1    red
3         1    orange
4         0    green
5         1    green

我需要找到所有可能的组合和出现的次数，所以我期望类似

1green    2
1red      1
1orange   1
0green    1

使用这个python: Combination of two Columns 我找到了所有可能的组合，但我不知道如何求和。或者有什么办法可以一步完成吗？

谢谢

Answer 1

试试：

pd.value_counts([*zip(df.output, df.colour)])

(1, green)     2
(0, green)     1
(1, orange)    1
(1, red)       1
dtype: int64

Answer 2

将Series.value_counts 用于连接列：

s = (df.output.astype(str) + df.colour).value_counts()
print (s)
1green     2
1red       1
0green     1
1orange    1
dtype: int64

或者可以使用GroupBy.size:

s = df.groupby(['output','colour']).size()
print (s)
output  colour
0       green     1
1       green     2
        orange    1
        red       1
dtype: int64

或DateFrame.value_counts

s = df[['output','colour']].value_counts()
print (s)
output  colour
1       green     2
0       green     1
1       orange    1
        red       1
dtype: int64

最后使用 DataFrame：

s = s.reset_index(name='count')

编辑：

对于所有组合使用：

s = df.groupby(['output','colour']).size().unstack(fill_value=0).stack()
print (s)
output  colour
0       green     1
        orange    0
        red       0
1       green     2
        orange    1
        red       1
dtype: int64

s = df[['output','colour']].value_counts().unstack(fill_value=0).stack()
print (s)
output  colour
0       green     1
        orange    0
        red       0
1       green     2
        orange    1
        red       1
dtype: int64