python - 如何将 csv 中的每一列保存到列表中？

Question

我正在尝试通过以下方式将每一列放入列表中:

f = open('file.csv', 'r') 
a = [float(item.split('.')[0]) for item in f]
b = [float(item.split('.')[1]) for item in f]

当我打印 a 时，我得到一个列表，其中包含第一列的所有值，但是当我打印 b 时，它显示一个空列表。我尝试将 a 的列更改为 b 的列，a 给了我我要的列，就在第一个定义的列表之后，它一直显示空列表。

Answer 1

您也可以只使用 csv.reader :

import csv

list_a = []
list_b = []

with open('file.csv', newline='') as csv_file:
    reader = csv.reader(csv_file, delimiter='.')
    next(reader, None)  # Skip the header.
    for a, b in reader:
        list_a.append(float(a))
        list_b.append(float(b))

---

或者使用 zip 技巧:

with open('file.csv', newline='') as csv_file:
    reader = csv.reader(csv_file, delimiter='.')
    next(reader, None)  # Skip the header.
    list_a, list_b = zip(*(map(float, row) for row in reader))