[英]Create a lists in a list variable in pandas DataFrame after groupby
我需要基于交易数据创建一个变量,该变量将是一个包含单个客户交易列表的列表
我设法创建了一个带有项目列表的变量:
dffg = pd.DataFrame(dff.groupby(["custid", "date", "transid"]).prod_sub.apply(lambda x: [x])).reset_index()
结果:
custid date transid prod_sub
1069 2001-02-03 00:00:00 1069_20010203 [[100101, 110117, 110108, 100314]]
1069 2001-02-10 00:00:00 1069_20010210 [[110217]]
1250 2001-02-04 00:00:00 1250_20010204 [[540110, 760687, 130317]]
1250 2001-02-10 00:00:00 1250_20010210 [[100109, 100205, 110411, 100102]]
4961 2001-02-05 00:00:00 4961_20010205 [[110504, 530101, 100422, 530108, 520437]]
我需要的是以下内容:
custid prod_sub
1069 [[100101, 110117, 110108, 100314], [110217]]
1250 [[540110, 760687, 130317], [100109, 100205, 110411, 100102]]
4961 [[110504, 530101, 100422, 530108, 520437]]
请帮忙
就像是
df.groupby('custid').prod_sub.apply(lambda x: x.tolist())
应该管用
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.