python -如何将大小为n的列表转换为大小为n/2 x n/2的二维numpy数组

Question

例如将3、1、2、4、4、2、3、1、1、3、4、2、2、4、1、3转换为

[
      [3, 1, 2, 4],
      [4, 2, 3, 1],
      [1, 3, 4, 2],
      [2, 4, 1, 3]
]

最短的方法...不将其转换为4个列表，然后转换为2d

Answer 1

在numpy中，您可以使用reshape命令将一维数组转换为二维数组。第一个参数是当前数组，第二个参数指定它的新“形状”应该是什么样子。在这种情况下，4行乘4列。

import numpy as np

my_array = [ 3, 1, 2, 4, 4, 2, 3, 1, 1, 3, 4, 2, 2, 4, 1, 3 ]
new_array = np.reshape(my_array, (4, 4))

将导致new_array为：

array([[3, 1, 2, 4],
       [4, 2, 3, 1],
       [1, 3, 4, 2],
       [2, 4, 1, 3]])

Akavall很友好，直接链接到reshape命令上的文档页面，复制到这里：https://docs.scipy.org/doc/numpy/reference/generated/numpy.reshape.html