熊猫数据框遍历行

Question

我有一个6000 * 5 dataframe ，如下所示：

 0      1      2     3     4
 0.001  0.020  0.502 0.773  0.332
 0.021  0.120  0.202 0.113  0.000
 0.001  0.420  0.432 0.003  0.123 ...

现在，我想循环浏览这6000行中的每行，并比较同一行中的值，以获取最大的元素并将列索引存储在新数据框中。 例如，对于第一行，最大元素应为0.773，第二行应为0.202，第三行为0.432。 所以我需要一个新的dataframe

3 2 2
...

如何解决这个问题？

Answer 1

您可以使用如下所示的idxmax（）。

df['max_col'] = df.idxmax(axis=1)

DataFrame.idxmax（axis = 0，skipna = True）

返回在请求轴上第一次出现最大值的索引。 不包括NA /空值。

http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.idxmax.html