首先,我不确定我是以最好的方式做到这一点,所以替代方案值得赞赏!
我有一个我创建的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()
很想知道,为什么并且有更好的方法?
答案 0 :(得分:0)
您可以使用nlargest
和reset_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