从 .csv 读取前 n 行并将列存储到列表中

时间:2021-02-25 22:12:19

标签: python pandas csv plot

我在植入 Python 代码时遇到问题,该代码从 n 文件读取第一个 rows .csv 并将 columns 的值存储在 {{ 1}}。列表的长度必须为 2000,该列表将用于创建 list

enter image description here

plot 文件中的 columns 未标记

2 个答案:

答案 0 :(得分:1)

您可以使用 pandas 来执行此操作:

import pandas as pd

df = pd.read_csv("test.csv", nrows=2000, header=None) #header = None avoids the first row to be read as column names
df_list = df.values.tolist()

答案 1 :(得分:0)

这可能会有所帮助,但为了将来参考,请阅读 Pandas 文档

import pandas as pd

df = pd.read_csv(<full_path_to_file>,nrows=2000,sep=<file_seperator>)
col_list = list(df.columns)