[英]How to merge rows of pandas DataFrame into list and transform this DataFrame into dict
我有一个DataFrame
col1 col2
0 a 1
1 b 2
2 c 3
3 a 4
4 b 5
我希望它转换如下:
col1 col2
0 a [1, 4]
1 b [2, 5]
2 c [3]
因此col1
的类似元素是唯一的,并且col2
相应值被合并到列表中。
那是第一部分。 第二个是:如何将这个DataFrame
导出到python dict
?
所以,结果,我希望得到以下的dict
{'a': ['1', '4'], 'b': ['2', '5'], 'c': ['3']}
合并值:
df.groupby('col1').col2.apply(list)
#col1
#a [1, 4]
#b [2, 5]
#c [3]
#Name: col2, dtype: object
转换为字典:
df.groupby('col1').col2.apply(list).to_dict()
# {'a': [1, 4], 'b': [2, 5], 'c': [3]}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.