Pandas:将csv文件作为列表加载
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Pandas:将csv文件作为列表加载相关的知识,希望对你有一定的参考价值。
customer gender customer_ids
1 0 [1,2,3]
2 1 [6,2,1]
3 0 [4,3,9]
我在上面的csv文件中有一些数据。我像这样加载csv文件
df = pd.read_csv('customer.csv', sep=' ')
这会将customer_ids加载为字符串,如"['1','2','3']", ...
但我需要像这样的numpy数组customer_ids数据
[list([1,2,3]), list([6,2,1]), list([4,3,9])]
答案
加载数据时,请指定converters
参数 -
df = pd.read_csv('customer.csv', sep=' ', converters={'customer_ids' : pd.eval})
df
customer gender customer_ids
0 1 0 [1, 2, 3]
1 2 1 [6, 2, 1]
2 3 0 [4, 3, 9]
df.customer_ids.tolist()
[[1, 2, 3], [6, 2, 1], [4, 3, 9]]
以上是关于Pandas:将csv文件作为列表加载的主要内容,如果未能解决你的问题,请参考以下文章