python索引值相加,python – 对多索引pandas DataFrame求和

您可以使用

transform,但获取重复值：

df['No. of classes'] = df.groupby(level='Class')['val'].transform('size')

print (df)

val No. of classes

Class subclass

A a 1 4

b 4 4

c 5 4

d 4 4

B a 1 2

b 2 2

但如果需要空值：

df['No. of classes'] = df.groupby(level='Class')

.apply(lambda x: pd.Series( [len(x)] + [np.nan] * (len(x)-1)))

.values

print (df)

val No. of classes

Class subclass

A a 1 4.0

b 4 NaN

c 5 NaN

d 4 NaN

B a 1 2.0

b 2 NaN

另一个获得最大数量的Class的解决方案是：

df = df.groupby(level=['Class'])

.apply(lambda x: x.index.get_level_values('subclass').nunique())

.idxmax()

print (df)

A