python - 如何在 python 3 中从 csv 创建自定义列表？

Question

这个问题已经有答案了:

Pandas DataFrame to List of Lists (14 个回答)

已关闭 5 年前。

我有如下所示的 csv 文件

Id,Date,P_Sales_Qty,Category
1,2016-03-20,1,17
2,2016-03-23,1,17
3,2016-03-25,2,17
4,2016-03-31,2,17
5,2016-04-02,3,17
6,2016-04-04,1,17
7,2016-04-09,1,17
8,2016-04-10,1,17
9,2016-04-28,1,17
10,2016-04-29,2,17
11,2016-04-30,1,17

我想创建一个列表，例如:

[2016-03-20 1   17
2016-03-23  1   17
2016-03-25  2   17
2016-03-31  2   17
2016-04-02  3   17
2016-04-04  1   17
2016-04-09  1   17
2016-04-10  1   17
2016-04-28  1   17
2016-04-29  2   17
2016-04-30  1   17]

请帮忙解决这个问题，我怎样才能实现这个目标？

Answer 1

只需为您的dataframe使用values属性即可。

list = df.values.tolist()
In[1] : list
Out[1]: [['2016-03-20', 1, 17],
         ..................
         ]

第二步是使用reduce方法来获得一个简单的列表。

from functools import reduce
list = reduce(lambda x, y: x+y, list)