python - 从带有索引的多列中选择最大值

Question

我有一个数据框

product     Jan     Feb     March   April
Apple       10      21      8       7
Banana      9       9       11      8
Orange      20      6       4       8

对于每个给定的产品，我想知道其各自月份的最大金额。

product     Month   Value       
Apple       Feb     21      
Banana      March   11
Orange      Jan     20

是否有任何方法可以在不显式遍历每一行的情况下执行此操作。

Answer 1

首先删除 assign 的列 product由 max 创建的新列和 idxmax :

a = df.drop('product', axis=1)
df =  df[['product']].assign(Month=a.idxmax(axis=1), Value=a.max(axis=1))
print (df)
  product  Month  Value
0   Apple    Feb     21
1  Banana  March     11
2  Orange    Jan     20