在python数据框中的每一行中查找最大值 [英] Finding highest values in each row in a data frame for python

问题描述

我想在每一行中找到最高的值，并在python中返回该值的列标题.例如，我想在每一行中找到前两个:

I'd like to find the highest values in each row and return the column header for the value in python. For example, I'd like to find the top two in each row:

df =  
       A    B    C    D  
       5    9    8    2  
       4    1    2    3  

我希望我的输出看起来像这样:

I'd like my for my output to look like this:

df =        
       B    C  
       A    D

推荐答案

您可以使用字典理解来在数据帧的每一行中生成largest_n值.我转置了数据框，然后将nlargest应用于每个列.我使用.index.tolist()提取所需的top_n列.最后，我将结果转换为所需的数据框.

You can use a dictionary comprehension to generate the largest_n values in each row of the dataframe. I transposed the dataframe and then applied nlargest to each of the columns. I used .index.tolist() to extract the desired top_n columns. Finally, I transposed this result to get the dataframe back into the desired shape.

top_n = 2
>>> pd.DataFrame({n: df.T[col].nlargest(top_n).index.tolist() 
                  for n, col in enumerate(df.T)}).T
   0  1
0  B  C
1  A  D

这篇关于在python数据框中的每一行中查找最大值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！