python - 我将如何通过二维数组“向上或向下移动”

Question

例如，给你一个数组：

array = [[2, 1, 4],
         [1, 3, 7],
         [7, 1, 4]]

并希望将每个垂直列打印为单独的列表：

res1 = [2, 1, 7]
res2 = [1, 3, 1]
res3 = [4, 7, 4]

对任何大小的二维数组进行编码的最有效方法是什么？

标签： pythonpython-3.x

Answer 1

如果您的 2D 数组很大并且需要对其进行大量计算，最好让其numpy处理

import numpy as np
array = np.array([[2, 1, 4],
                  [1, 3, 7],
                  [7, 1, 4]])

for col in array.T:
    print(col)