使用行名,列名和最大列值创建数据框

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用行名,列名和最大列值创建数据框相关的知识,希望对你有一定的参考价值。

我正在浏览我从PACKT购买的视频包以学习熊猫。作者使用jijna2 style()突出显示每列中的最大值。我很快发现我不能在PyCharm中使用这种技术。所以我决定提取这些值。

我想要做的是通过从具有N列的数据帧中提取行索引,列名和最大列值来创建三列数据帧,然后创建新的数据帧。新数据框将显示每一行(如果有关系显示所有适当的行),列和该列中的最大值。

我创建了一个玩具数据框,只是为了完成代码的工作。

我的代码在下面,连同输出,在最底层,是我真正想要新数据帧的样子。

我知道我正在使用打印声明。到目前为止,该代码是我唯一能够正确选取多行的代码。

我抓住了整排,这是我不想要的。我也不确定如何从提取的数据构造建议的新数据帧。

import pandas as pd


raw_data = {
            'dogs': [42, 39, 86, 15, 23, 57, 68, 81, 86],
            'cats': [52, 41, 79, 80, 34, 47, 19, 22, 59],
            'sheep': [62, 37, 84, 51, 67, 32, 23, 89, 73],
            'lizards': [72, 43, 36, 26, 53, 88, 88, 34, 69],
            'birds': [82, 35, 77, 63, 18, 12, 45, 56, 58],
            }

df = pd.DataFrame(raw_data,
                  index=pd.Index(['row_1', 'row_2', 'row_3', 'row_4', 'row_5', 'row_6', 'row_7', 'row_8', 'row_9'], name='Rows'),
                  columns=pd.Index(['dogs', 'cats', 'sheep', 'lizards', 'birds'], name='animals'))

print(df)
print()

# Get a list of all columns names
cols = df.columns
print(cols)
print('*****')

for col in cols:
    print((df[df[col] == df[col].max()]))


'''
animals  dogs  cats  sheep  lizards  birds
Rows                                      
row_3      86    79     84       36     77
row_9      86    59     73       69     58
animals  dogs  cats  sheep  lizards  birds
Rows                                      
row_4      15    80     51       26     63
animals  dogs  cats  sheep  lizards  birds
Rows                                      
row_8      81    22     89       34     56
animals  dogs  cats  sheep  lizards  birds
Rows                                      
row_6      57    47     32       88     12
row_7      68    19     23       88     45
animals  dogs  cats  sheep  lizards  birds
Rows                                      
row_1      42    52     62       72     82
'''

row_3     dogs        86
row_9     dogs        86
row_4     cats        80
row_8     sheep       89
row_6     lizards     88
row_7     lizards     88
row_1     birds       82
答案

使用numpy.where作为匹配的maxes的索引并通过索引创建新的DataFrame - 如果在大型DataFrame中性能很重要,那就更好了:

c, r = np.where(df.eq(df.max()).T)
df = pd.DataFrame({'idx':df.index[r], 'cols':df.columns[c], 'vals': df.values[r, c]})
print(df)
     idx     cols  vals
0  row_3     dogs    86
1  row_9     dogs    86
2  row_4     cats    80
3  row_8    sheep    89
4  row_6  lizards    88
5  row_7  lizards    88
6  row_1    birds    82

使用DataFrame.unstackGroupBy.transform的另一个唯一的熊猫解决方案,用于比较第一级每组的max值:

s = df.unstack()
df = s[s.groupby(level=0).transform('max').eq(s)].reset_index(name='vals')
print(df)
   animals   Rows  vals
0     dogs  row_3    86
1     dogs  row_9    86
2     cats  row_4    80
3    sheep  row_8    89
4  lizards  row_6    88
5  lizards  row_7    88
6    birds  row_1    82
另一答案

您可以使用.where和eq到mask检查no max值作为NaN然后stack

df.where(df.eq(df.max())).stack().sort_index(level=1).reset_index()
    Rows  animals     0
0  row_3     dogs  86.0
1  row_9     dogs  86.0
2  row_4     cats  80.0
3  row_8    sheep  89.0
4  row_6  lizards  88.0
5  row_7  lizards  88.0
6  row_1    birds  82.0

以上是关于使用行名,列名和最大列值创建数据框的主要内容,如果未能解决你的问题,请参考以下文章

R中的列名和行名匹配

Pandas - 在应用映射期间检索每个元素的行名和列名

如何根据行名和列名的比较构建方阵

根据现有的列名和列值在 python 数据框中创建列

将字典转换为数据框,键作为列名,键值作为数据框的列值

从数据框中删除行名的显示