Pandas 读取 csv 得 DataFrame 转换成 List
DataFrame转为list用 pandas 读取 csv 文件时,会得到 DataFrame,后期如果需要进行一些 List 相关操作,需要进行转换,分为三个步骤:用 np.array 将DataFrame 转换成 np.ndarray对 ndarray 进行 reshape并用 tolist 转换成 List取 List 的第一个元素示例代码如下:import pandas ...
·
DataFrame转为list
用 pandas 读取 csv 文件时,会得到 DataFrame,后期如果需要进行一些 List 相关操作,需要进行转换,分为三个步骤:
- 用 np.array 将DataFrame 转换成 np.ndarray
- 对 ndarray 进行 reshape并用 tolist 转换成 List
- 取 List 的第一个元素
示例代码如下:
import pandas as pd
import numpy as np
corpus = pd.read_csv('corpus.csv') # 得到 DataFrame
corpus = np.array(corpus) # 转换为 ndarray [[1], [2], [3]]
corpus = corpus.reshape(1, len(corpus)).tolist() # 转换成 List [[1, 2, 3]]
corpus = corpus[0] # 取第一个元素得到最终结果 [1, 2, 3]
啊,今天小伙伴看到之后告诉我还有更简单的一种办法,我果然还是太年轻了,果然还是个弟弟。
corpus = pd.read_csv('corpus.csv')
corpus = corpus.values.tolist()
也就是说可以直接利用 Dataframe.values.tolist() 转换成列表。再次感慨,我真是个弟弟。(长叹一口气)
更多推荐
已为社区贡献1条内容
所有评论(0)