我有一个PandasDataFrame
如下:
df = pd.DataFrame({'id' : [1,1,1,2,2,3,3,3,3,4,4,5,6,6,6,7,7],
'value' : ["first","second","second","first",
"second","first","third","fourth",
"fifth","second","fifth","first",
"first","second","third","fourth","fifth"]})
我想按["id","value"]
分组,并获取每组的第一行:
id value
0 1 first
1 1 second
2 1 second
3 2 first
4 2 second
5 3 first
6 3 third
7 3 fourth
8 3 fifth
9 4 second
10 4 fifth
11 5 first
12 6 first
13 6 second
14 6 third
15 7 fourth
16 7 fifth
预期成果:
id value
1 first
2 first
3 first
4 second
5 first
6 first
7 fourth
我试着跟随,它只给出了DataFrame
的第一行。任何关于这方面的帮助都很感激。
In [25]: for index, row in df.iterrows():
....: df2 = pd.DataFrame(df.groupby(['id','value']).reset_index().ix[0])
8条答案
按热度按时间djmepvbi1#
如果需要
id
作为列:要获取前n个记录,可以使用head():
pieyvz9o2#
这将为您提供每组的第二行(零索引,
nth(0)
与first()
相同):文件:www.example.comhttp://pandas.pydata.org/pandas-docs/stable/groupby.html#taking-the-nth-row-of-each-group
anhgbhbe3#
如果需要获取第一行,我建议使用
.nth(0)
而不是.first()
。它们之间的区别在于处理NaN的方式,因此
.nth(0)
将返回group的第一行,无论该行中的值是什么,而.first()
最终将返回每列中的第一个notNaN
值。例如,如果您的数据集是:
还有
zxlwwiss4#
如果你只需要每组的第一行,我们可以使用
drop_duplicates
,注意函数的默认方法keep='first'
。3phpmpom5#
也许这就是你想要的
lc8prwob6#
我想“first”意味着您已经按照自己的意愿对DataFrame进行了排序。
我做的是:
df.groupby('id ').agg('first')我想“first”意味着你已经按照自己的意愿对DataFrame进行了排序。
好的方面是你可以插入任何你想要的函数:
输出数据框具有多索引列
vwoqyblh7#
考虑到
'id'
列是数值类型,例如int32
/int64
,也可以如下使用groupby.rank()
如果要重置索引,只需传递
.reset_index()
,例如如果不需要
index
和id
列3okqufwl8#
可以使用方法
take
,该方法接受要选择的元素的索引列表: