如何将 csv 文件的第二列转换为浮点数列表？

Question

我有这样的 csv 文件：

string, 3.54545,4.3434,3.34435543
string, 4.54545,67.3434,5.34435543
...
stringN, 5.54545,1.3434,9.34435543

如何使用 csv 模块提取两个不同列表中的第一列（字符串）和第二列（浮点数）？例如，我想得到这样的东西：

list1 = [string,string,...,string]
list2 = [3.54545,4.54545,..,5.54545]

其中list1 是字符串列表，list2 是浮点数列表。我用 pandas 尝试了以下操作，问题是它需要大量读取文件：

df = pd.read_csv('test_dict.csv', header = None)

list1 = df[0].values.tolist()
list2 = df[1].values.tolist()

提前谢谢各位！

Answer 1

如果你想要前两列，你可以压缩：

import  csv

with open("in.csv") as f:
    reader = csv.reader(f)
    zipped = zip(*reader)
    s, f = list(next(zipped)), list(map(float, next(zipped)))

对于 python 2 使用 itertools.izip：

import  csv
from itertools import  izip

with open("in.csv") as f:
    reader = csv.reader(f)
    zipped = izip(*reader)
    s, f = list(next(zipped)), map(float, next(zipped))
    print(s, f)
 (['string', 'string', 'stringN'], [3.54545, 4.54545, 5.54545])

Answer 2

你可以像这样使用 csv 模块来做到这一点，但正如我在我的 cmets 中所说，不要指望它比使用 pandas 更快。

import csv

col1 = []
col2 = []

with open('test_dict.csv') as f:
  for row in csv.reader(f):
    col1.append(row[0])
    col2.append(row[1])