python - Pandas 如何选择行具有最大值的列？

Question

假设我有以下数据框:

df = pd.DataFrame({'A':[0.1,0.1,0.2],
                  'B': [0.22,0.2,0.22],
                  'C': [0.3, 0.2,0.333]})

df.index = list('abc')
df
     A     B      C
a  0.1  0.22  0.300
b  0.1  0.20  0.200
c  0.2  0.22  0.333

如何选择索引b具有最大值的所有列。

此处，索引在 B 列和 C 列中的最大值为 2。所以，所需的答案是:

required_answer = ['B','C']

我的尝试:

df[df.loc['b'] ==df.loc['b'].max()]

Answer 1

你可以试试这个:

df.columns[df.loc['b'] == df.loc['b'].max()].tolist()
# ['B', 'C']