python - 有效地将 CSV 列解压到单独的列表中

Question

我正在优化我的脚本并发现这个问题:

这里我有一个 csv 文件，其中第一列只是索引，第二列包含一个字符串(任意长度的句子)。我想创建两个变量“index”和“string”，分别包含所有索引和字符串。这是我的代码:

with open(file_name, 'r', encoding="utf8") as csvfile:
    train_set_x = csv.reader(csvfile, delimiter=',', quotechar='|')
    index = [[c[0],c[1]] for c in train_set_x]
    text = [a[1] for a in index]

这可以完成工作，但是需要 2 次迭代。我问是否有更干净的方法来做到这一点？谢谢

Answer 1

肯定有。使用 zip 进行可迭代解包。

index, text = zip(*((c[0], c[1]) for c in train_set_x))

<小时/>

MCVE:

In [152]: x, y = zip(*[(1, 2), (3, 4), (5, 6)])

In [153]: x
Out[153]: (1, 3, 5)

In [154]: y
Out[154]: (2, 4, 6)