python - csv.DictReader() 是否返回字典对象？

Question

当我尝试使用 csv.DictReader() 将 CSV 文件传输到字典时，如下所示:

csv_file = open(input_file, "r")
data = csv.DictReader(csv_file)

但是，我不能将任何字典方法应用于数据。

因此，我打印了数据的类型。

print(type(data))

并找到类型是:

<class 'csv.DictReader'>

那么 csv.DictReader() 真的返回一个字典对象吗？如何从 CSV 获取字典对象？

我的 CSV 文件如下所示:

[['play', 'weather', 'temperature'], ['yes', 'sunny', '77'], ['no', 'rainny', '60'], ['yes', 'windy', '70'],...]

实际上，我没有预期的输出。我只想存储这些值以供我进一步计算。

编辑(根据评论): 输出可能分为两部分。第一个是 csv 文件的 header ，例如 ['play', 'weather', 'temperature'] 。第二个可以是这样的:[['yes', 'sunny', '77'], ['no', 'rainny', '60'], ['yes', 'windy', '70 ']...]

Answer 1

csv.reader

鉴于您想要的输出，您不需要使用 dict 或 csv.DictReader。相反，只需使用 csv.reader，它返回一个迭代器。然后使用next和list分别提取headers和data:

from io import StringIO
import csv

x = StringIO("""play,weather,temperature
yes,sunny,77
no,rainy,60
yes,windy,70""")

# replace x with open('file.csv', 'r')
with x as fin:
    reader = csv.reader(fin)
    headers = next(reader)  # get headers from first row
    data = list(reader)     # exhaust iterator from second row onwards

结果是一个标题列表和一个数据列表列表:

print(headers)

['play', 'weather', 'temperature']

print(data)

[['yes', 'sunny', '77'],
 ['no', 'rainy', '60'],
 ['yes', 'windy', '70']]

Pandas

如果您愿意使用第三方库，Pandas 可能是更好的选择，因为它可以更方便地处理类型转换和索引:

import pandas as pd

df = pd.read_csv('file.csv')

结果是一个pd.DataFrame对象:

print(df)

  play weather  temperature
0  yes   sunny           77
1   no   rainy           60
2  yes   windy           70

print(type(df))

<class 'pandas.core.frame.DataFrame'>