python - 提取满足给定条件的单元格的列标签

Question

假设手头的数据是以下形式:

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 A C。

感谢您的任何建议。

Answer 1

你可以使用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)