Pandas 如何在 select 列中一行具有最大值？

Question

假设我关注 dataframe：

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

如何 select 索引 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']

Answer 2

我给你几个选择。 第一个是最好的：

[*df.columns[df.loc['b'].eq(df.loc['b'].max())]]

['B', 'C']

---

df.loc[:,df.loc['b'].eq(df.loc['b'].max())].columns.tolist()

['B', 'C']

---

[*df.T.index[df.loc['b'].eq(df.loc['b'].max())]]

['B', 'C']