Python 中的 CSV 列到数组

我很难处理一个在我看来很简单的问题。我正在尝试导入 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 列的方法感兴趣。

你需要改变column1.extend(row[0]) to column1.append(row[0])（显然，第 2 列也是如此）。 Extend 用于将一个列表的内容添加到另一个列表，append 用于添加单个元素。 Extend 告诉 python 将字符串视为其字符列表并附加每个字符。

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