在列表列表中，有没有办法同时获取元素的行和列中的元素？ [复制]

Question

我有一个名为square 的简单列表：

square = [[1,2,3],[4,5,6],[7,8,9]]

我的目标是遍历九个元素中的每一个，并同时获取该元素的行和列中的所有值。在此示例中，第一次迭代（对于元素 1）应该返回 [1,2,3] 和 [1,4,7]，第二次（对于元素 2）将返回 [1,2,3] 和 [2,5,8]，等等。以下代码有效：

for r in range(3):
    for c in range(3):
        row = square[r]
        col = [square[c1][c] for c1 in range(3)]

但是是否有另一种方法使用 base Python 来执行此操作？

我可以使用for c in zip(*square): 对列表进行转置和迭代，如果我只需要列一次，这可以工作，但是没有办法使用数组切片来索引列而不存储矩阵的转置以及矩阵本身？

不幸的是，目前无法选择像 numpy 这样的库，因为它们需要首先通过我们的代码审查流程，这需要很长时间（大约六个月。 ..)。我没有编写政策或设计程序，但这是我必须在这个机构工作的政策。我已经提交了 numpy 的请求，但与此同时，我只是在使用基础 Python。

Answer 1

如果您想使用标准 Python 执行此操作，您可能需要尝试转置列表。然后，您可以正常索引。您可以使用zip(*square) 后跟map(list,) 将其转回列表：

square = [[1,2,3],[4,5,6],[7,8,9]]

square_transposed = map(list, zip(*square))

然后square[1] 会得到你[2,5,8]。

如果您想知道 * 运算符在此上下文中的作用，it unpacks the list to a series of position arguments。所以 zip 看到[1,2,3] [4,5,6] [7,8,9] 并将它们压缩在一起作为[(1, 4, 7), (2, 5, 8), (3, 6, 9)]。然后最终的映射将元组转换为列表。

Answer 2

使用 numpy

>>> import numpy as np
>>> square = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> square[0]
array([1, 2, 3])
>>> square[..., 0]
array([1, 4, 7])
>>> square[:, 0]
array([1, 4, 7])

仅使用 Python

>>> square = [[1,2,3],[4,5,6],[7,8,9]]
>>> square[0]
[1, 2, 3]
>>> zip(*square)[0]
(1, 4, 7)