追加到Python中行/列长度未知的二维列表

Question

我想要翻转二维列表上的行和列，但遇到了麻烦。 Python 不允许将值分配给未初始化的“单元格”。我不知道如何初始化我的网格，因为我不知道提前的长度。以下内容：

data = list()
maReader = csv.reader(open('TEST.csv', 'rb'), delimiter=',', quotechar='"')
for rindex, row in enumerate(maReader):
    for iindex, item in enumerate(row):
        data[iindex][rindex] = item

失败并出现“IndexError：列表分配索引超出范围”错误。我可以循环两次，第一次计算列和行，然后初始化网格，然后再次循环并赋值，但这似乎非常浪费。

我对 Python 还很陌生，并且认为有一种我缺少的简单方法可以做到这一点。

感谢您的帮助。

Answer 1

这是与 @retracile 的 LBYL 方法。

data = [] #list()
maReader = csv.reader(open('TEST.csv', 'rb'), delimiter=',', quotechar='"')
for rindex, row in enumerate(maReader):
    for iindex, item in enumerate(row):
        try:
            data[iindex].append(item)
        except IndexError:
            data.append([item])

Answer 2

假设所有行的长度相等，您可以这样处理：

data = list()
maReader = csv.reader(open('TEST.csv', 'rb'), delimiter=',', quotechar='"')
for rindex, row in enumerate(maReader):
    for iindex, item in enumerate(row):
        if len(data) <= iindex:
            data.append([])
        data[iindex].append(item)

Answer 3

这应该可以：

maReader = csv.reader(open('TEST.csv', 'rb'), delimiter=',', quotechar='"')
data = zip(*maReader)

不过，zip 给出了一个元组列表。如果您需要列表列表，您可以执行以下任一操作：

data = [list(x) for x in zip(*maReader)]
data = map(list, zip(*maReader))

Answer 4

如果所有行的长度相等，则 data = zip(*maReader) 应该这样做。
如果不是：

from itertools import izip_longest

list(izip_longest(*f, fillvalue=''))

另请参阅Python 中的矩阵转置。