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() 转换成列表。再次感慨,我真是个弟弟。(长叹一口气)

Logo

助力广东及东莞地区开发者,代码托管、在线学习与竞赛、技术交流与分享、资源共享、职业发展,成为松山湖开发者首选的工作与学习平台

更多推荐