python - 返回具有其中一个轴最大值的 pandas DataFrame 的所有条目？

Question

我确定这是一个如此基本的问题，但我似乎无法为这个确切的问题找到任何解决方案......

我有一个如下所示的 DataFrame，我需要找到“B”的最大值的条目。在下面的情况下，我希望条目 2 和 3 的 B = 25。任何返回条目索引或整个条目的解决方案都可以。

Name: 6417, dtype: float64
          A     B     C
0      8.00     0     0
1     24.00    20     5
2     31.20    25    10
3     30.00    25    15
4      3.20    20    20

注意：我看过一些关于使用 pd.idxmax() 的帖子，但这只会返回一个条目的索引，即使给定轴的最大值有多个条目也是如此。

干杯

标签： pythonpandas

Answer 1

我会这样做：

def function(dataframe):

    list_max, max = [], 0

    for row in dataframe.itertuples():

        if row.B > max:        
            max = row.B
            list_max.clear()
            list_max.append(row.Index)

        elif row.B = max:
            list_max.append(row.Index)

    return list_max, max