📜  pandas 为每行查找具有最大值的列 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:50.900000             🧑  作者: Mango

代码示例2
# credit to Stack Overflow in the source link

df.head()
Communications and Search   Business    General Lifestyle
0   0.745763    0.050847    0.118644    0.084746
0   0.333333    0.000000    0.583333    0.083333
0   0.617021    0.042553    0.297872    0.042553
0   0.435897    0.000000    0.410256    0.153846
0   0.358974    0.076923    0.410256    0.153846

df.idxmax(axis=1)
0    Communications
1          Business
2    Communications
3    Communications
4          Business
dtype: object