我有一个像这样的DataFrame:
In [7]: frame.head() Out[7]: 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
在这里,我想问一下如何获取每行具有最大值的列名,所需的输出是这样的:
In [7]: frame.head() Out[7]: Communications and Search Business General Lifestyle Max 0 0.745763 0.050847 0.118644 0.084746 Communications 0 0.333333 0.000000 0.583333 0.083333 Business 0 0.617021 0.042553 0.297872 0.042553 Communications 0 0.435897 0.000000 0.410256 0.153846 Communications 0 0.358974 0.076923 0.410256 0.153846 Business
您可以使用idxmaxwithaxis=1来查找每行中具有最大值的列:
idxmax
axis=1
>>> df.idxmax(axis=1) 0 Communications 1 Business 2 Communications 3 Communications 4 Business dtype: object
要创建新列“Max”,请使用df['Max'] = df.idxmax(axis=1).
df['Max'] = df.idxmax(axis=1)
要查找每列中出现最大值的 行df.idxmax()索引,请使用(或等效地df.idxmax(axis=0))。
df.idxmax()
df.idxmax(axis=0)