如何在Python中使用csv.DictReader将CSV指定列数据保存为列表

首先得说，你之前的代码之所以得到字符拆分的列表，是因为list(row["col1"])会把单个字符串（比如"house"）当成可迭代对象来处理，遍历它自然就会拆成每个字符啦。要拿到整列的字符串列表，咱们换个思路就行：

正确实现方式 #

先初始化一个空列表，然后在循环里把每行的col1值完整追加进去：

import csv

# 先创建一个空列表用来存数据
col1_values = []
with open("house.csv", "r+", newline="", encoding="UTF-8") as csv_file:
    file_r = csv.DictReader(csv_file)
    for row in file_r:
        # 把整行的col1字符串直接加到列表里
        col1_values.append(row["col1"])

print(col1_values)  # 输出结果就是 ['house', 'cottage']

更简洁的写法（列表推导式） #

如果喜欢紧凑的代码，也可以用列表推导式一步到位：

import csv

with open("house.csv", "r+", newline="", encoding="UTF-8") as csv_file:
    file_r = csv.DictReader(csv_file)
    col1_values = [row["col1"] for row in file_r]

print(col1_values)

这个写法和上面循环追加的效果完全一样，只是把循环逻辑浓缩成了一行。

内容的提问来源于stack exchange，提问作者Ciro Longobardi