Pandas 读取 csv 得 DataFrame 转换成 List

DataFrame转为list

用 pandas 读取 csv 文件时，会得到 DataFrame，后期如果需要进行一些 List 相关操作，需要进行转换，分为三个步骤：

1. 用 np.array 将DataFrame 转换成 np.ndarray
2. 对 ndarray 进行 reshape并用 tolist 转换成 List
3. 取 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() 转换成列表。再次感慨，我真是个弟弟。（长叹一口气）