![](/img/trans.png)
[英]How to groupby by a column and return the values of other columns as lists in pandas?
[英]Pandas: groupby column A and make lists of tuples from other columns?
我想将用户交易汇总到 Pandas 中的列表中。 我不知道如何制作包含多个字段的列表。 例如,
df = pd.DataFrame({'user':[1,1,2,2,3],
'time':[20,10,11,18, 15],
'amount':[10.99, 4.99, 2.99, 1.99, 10.99]})
看起来像
amount time user
0 10.99 20 1
1 4.99 10 1
2 2.99 11 2
3 1.99 18 2
4 10.99 15 3
如果我这样做
print(df.groupby('user')['time'].apply(list))
我得到
user
1 [20, 10]
2 [11, 18]
3 [15]
但如果我这样做
df.groupby('user')[['time', 'amount']].apply(list)
我得到
user
1 [time, amount]
2 [time, amount]
3 [time, amount]
感谢下面的答案,我知道我可以做到这一点
df.groupby('user').agg(lambda x: x.tolist()))
得到
amount time
user
1 [10.99, 4.99] [20, 10]
2 [2.99, 1.99] [11, 18]
3 [10.99] [15]
但我想以相同的顺序对时间和金额进行排序 - 这样我就可以按顺序浏览每个用户的交易。
我正在寻找一种方法来产生这个:
amount-time-tuple
user
1 [(20, 10.99), (10, 4.99)]
2 [(11, 2.99), (18, 1.99)]
3 [(15, 10.99)]
但也许有一种方法可以在不“重复”两列的情况下进行排序?
apply(list)
将考虑系列索引而不是值。我认为您正在寻找
df.groupby('user')[['time', 'amount']].apply(lambda x: x.values.tolist())
user 1 [[23.0, 2.99], [50.0, 1.99]] 2 [[12.0, 1.99]]
为时间元组atpair
一个新列
df['atpair'] = list(zip(df.amount, df.time))
数据框看起来像
user time amount atpair
0 1 20 10.99 (10.99, 20)
1 1 10 4.99 (4.99, 10)
2 2 11 2.99 (2.99, 11)
3 2 18 1.99 (1.99, 18)
4 3 15 10.99 (10.99, 15)
现在执行 groupby 并将列表附加到atpair
df = df.groupby('user')['atpair'].apply(lambda x : x.values.tolist())
数据框看起来像
user
1 [(10.99, 20), (4.99, 10)]
2 [(2.99, 11), (1.99, 18)]
3 [(10.99, 15)]
IUC:
In [101]: df.groupby('user').agg(lambda x: x.tolist())
Out[101]:
time amount
user
1 [23, 50] [2.99, 1.99]
2 [12] [1.99]
基于巴拉特的回答
df.groupby('user')[['time', 'amount']].apply(lambda x: list(map(tuple,x.values)))
这可以得到:
user
1 [(20.0, 10.99), (10.0, 4.99)]
2 [(11.0, 2.99), (18.0, 1.99)]
3 [(15.0, 10.99)]
dtype: object
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.