如何使用NumPy创建自定义4x4阵列？

Question

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

我知道使用np.zeros((4,4))输出一个全0的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]])