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文件作为列表加载的主要内容,如果未能解决你的问题,请参考以下文章

如何使用pandas将csv列作为dtype列表读取?

将包含长列表的 Pandas df 保存为 csv 文件

Python:将列表写入 Pandas 中的列

如何将csv数据加载到matplotlib?

如何自动将csv转换为pandas?

如何将 CSV/TSV 文件从 Pig 加载/导出到 Pandas?