python - 如何使用 NumPy 创建自定义 4x4 数组？

Question

我是 Python 新手，我在使用数组函数时遇到了一些麻烦。我想制作一个 4 x 4 数组，其中包含从 1 到 16 的数字。

我知道使用 np.zeros((4,4))输出一个全为零的 4x4 数组。使用np.array(range(17))我可以获得所需数字的数组，但形状不正确（4x4）。

一定很简单吧？非常感谢所有评论。

Answer 1

问题是您正在创建一个包含 17 个值（从 0 到 16）的数组，无法将其重新整形为 4x4。反而：

>>> a = np.arange(1, 17).reshape(4,4)
>>> a
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12],
       [13, 14, 15, 16]])