分组和排序后:为什么我的python数据帧列名称行,而不是单行?

时间:2016-01-12 10:11:17

标签: python sorting pandas grouping

首先,我不确定我是以最好的方式做到这一点,所以替代方案值得赞赏!

我有一个我创建的python pandas数据框: raw_data = {'Name' : pd.Series(['david','andrew','calvin','david','calvin','david'], index=['a', 'b', 'c', 'd','e','f']),'Age' : pd.Series([35, 40, 5, 35, 5, 35], index=['a', 'b', 'c', 'd', 'e','f'])} df = pd.DataFrame(raw_data, columns = ['Name','Age'])

我现在想要分组并进行排序(根据名称'出现次数的频率)并返回前2条记录。即以下输出:

  Name Count
david      3
calvin     2

我首先添加一个" Count"柱:

df = pd.DataFrame(raw_data, columns = ['Name','Age'])

给出:

     Name   Age Count
a   david   35  3
b   andrew  39  1
c   calvin  5   2
d   david   34  3
e   calvin  5   2
f   david   35  3

然后,我可以使用以下代码对其进行分组和排序:

df[['Name','Count']].groupby('Name').count().sort_values('Count',ascending=0).head(2)

这就是我遇到问题的地方,数据是"正确",但列标题似乎在不同的行中...请参阅屏幕截图: Column Header Alignment problem pic

我有3个问题

1)为什么会这样?

2)我该如何解决?

3)有更好的方法吗?

谢谢!

[编辑] - 我已回答问题2,(如何解决)。显然我需要重置索引:

df.reset_index()

很想知道,为什么并且有更好的方法?

1 个答案:

答案 0 :(得分:0)

您可以使用nlargestreset_index

print df

#     Name  Age
#a   david   35
#b  andrew   40
#c  calvin    5
#d   david   35
#e  calvin    5
#f   david   35

df1 =  df.groupby('Name').apply(len)
print df1

#Name
#andrew    1
#calvin    2
#david     3
#dtype: int64
print df1.nlargest(2).reset_index(name='Count')

#     Name  Count
#0   david      3
#1  calvin      2