python pandas 条件求和_python - 如何对多个列的组进行求和或计数_pandas_酷徒编程知识库...

我試圖分組對pandas dataframe中的行進行計數或求和

例如，我有以下dataframe：import numpy as np

df = pd.DataFrame(np.random.randint(0,5,size=(5, 7)), columns=["grey2","red1","blue1","red2","red3","blue2","grey1"])

grey2 red1 blue1 red2 red3 blue2 grey1

0 4 3 0 2 4 0 2

1 4 2 0 4 0 3 1

2 1 1 3 1 1 3 1

3 4 4 1 4 1 1 1

4 3 4 1 0 3 3 1

我想在這個分組，例如，所有列按顏色分組，我期望的是：

如果我把這些數字相加blue 15

grey 22

red 34

如果我計數(x> 0)我就會得到blue 7

grey 10

red 13

pd.pivot_table(data=df, index=df.index, values=["red1","red2","red3"], aggfunc='sum', margins=True)

red1 red2 red3

0 3 2 4

1 2 4 0

2 1 1 1

3 4 4 1

4 4 0 3

ALL 14 11 9

pd.pivot_table(data=df, index=df.index, values=["red1","red2","red3"], aggfunc='count', margins=True)

red1 red2 red3

0 1 1 1

1 1 1 1

2 1 1 1

3 1 1 1

4 1 1 1

All 5 5 5

注意：在此例子中，我僅使用顏色來簡化情況，但是我可能有很多列，它們被稱為col001至col300等，所以這些組可能是：blue = col131, col254, col005

red = col023, col190, col053

等等。