Python：获取 pandas 数据帧每行的最大值列
Python: Get Columns of max values each row of an pandas dataframe
在python中获取pandas数据帧矩阵（没有索引列）中每行最大值的列号的最佳方法是什么。例如。如果我有
Date
Company 1
Company 2
Company 3
01.01.2020
23
21
14
02.01.2020
22
12
22
03.01.2020
11
11
12
....
...
...
...
02.01.2020
2
14
3
输出应该是向量：
[1, 1, 3, ..., 2]
使用idxmax:
In [949]: cols = df.iloc[:, 1:].idxmax(1).tolist()
In [950]: cols
Out[950]: ['Company 1', 'Company 1', 'Company 3']
如果你想要column index:
In [951]: [df.columns.get_loc(i) for i in cols]
Out[951]: [1, 1, 3]
在python中获取pandas数据帧矩阵（没有索引列）中每行最大值的列号的最佳方法是什么。例如。如果我有
| Date | Company 1 | Company 2 | Company 3 |
|---|---|---|---|
| 01.01.2020 | 23 | 21 | 14 |
| 02.01.2020 | 22 | 12 | 22 |
| 03.01.2020 | 11 | 11 | 12 |
| .... | ... | ... | ... |
| 02.01.2020 | 2 | 14 | 3 |
输出应该是向量：
[1, 1, 3, ..., 2]
使用idxmax:
In [949]: cols = df.iloc[:, 1:].idxmax(1).tolist()
In [950]: cols
Out[950]: ['Company 1', 'Company 1', 'Company 3']
如果你想要column index:
In [951]: [df.columns.get_loc(i) for i in cols]
Out[951]: [1, 1, 3]