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

Question

我是Python的新手,我在使用数组函数时遇到了一些麻烦.我想制作一个4乘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]])