python - CSV 列到 Python 中的数组

Question

我很难处理在我看来很简单的问题。我正在尝试导入一个 csv 并将其列拆分为我可以运行不同操作的数组，然后将 zip() 组合在一起。

import csv

data = csv.reader(open('test.csv', 'rb'), delimiter=",", quotechar='|')
column1, column2 = [], []

for row in data:
    column1.extend(row[0])
    column2.extend(row[1])

print column1
print column2

此代码打印两个数组，其元素是单独的字符而不是字符串。当我尝试对单个列执行此操作时，column1.extend(row) 会执行我想要的操作。

我对解决这个特定问题或将其推广到 n 列的方法很感兴趣。

Answer 1

您需要将 column1.extend(row[0]) 更改为 column1.append(row[0])(很明显，column2 也是如此)。 Extend 用于将一个列表的内容添加到另一个列表，append 用于添加单个元素。 Extend 告诉 python 将字符串视为其字符列表并附加每个字符。

>>> lst = []
>>> lst.extend("foo")
>>> lst
['f', 'o', 'o']
>>> lst.append("foo")
>>> lst
['f', 'o', 'o', 'foo']