假设手头的数据具有以下格式:
import pandas as pd
df = pd.DataFrame({'A':[1,10,20], 'B':[4,40,50], 'C':[10,11,12]})
我可以按以下方式计算最小值:
df.min(axis=1)
返回1 10 12.
除了值之外,我想创建一个包含相应单元格列标签的pandas系列.
也就是说,我想获得A AC.
感谢您的任何建议.
解决方法:
您可以使用idxmin(axis=1)方法:
In [8]: df.min(axis=1)
Out[8]:
0 1
1 10
2 12
dtype: int64
In [9]: df.idxmin(axis=1)
Out[9]:
0 A
1 A
2 C
dtype: object
In [11]: df.idxmin(axis=1).values
Out[11]: array(['A', 'A', 'C'], dtype=object)