我pandas
可以选择执行以下操作来获取包含每行最大值的列的名称:
import pandas as pd
df = pd.DataFrame({'a': [1, 2, 3, 4, 5], 'b': [5, 4, 3, 2, 1]})
df['Largest'] = df.idxmax(axis=1)
这让我明白:
A | 乙 | 最大 | |
---|---|---|---|
0 | 1 | 5 | 乙 |
1 | 2 | 4 | 乙 |
2 | 3 | 3 | A |
3 | 4 | 2 | A |
4 | 5 | 1 | A |
我怎样才能在 中进行这种操作polars
?似乎没有idxmax
方法,并且max_horizontal
似乎只返回值而不是任何索引信息。