使用for循环的Numpy 2D数组将其用数字的上升顺序填充

Question

也有类似的问题，但它们不是python或b(不使用我的特定结构使用以这种方式使用2D numpy阵列的for lip loop。

我想用数字1,2,3,4,5等填充以下numpy阵列

目前它从1-10起填充每一行，然后再次开始：

import numpy
a = numpy.zeros((16,11))
for i in range(11):
  for j in range(16):
    for k in range(16):
      a[j,i]=i+1

print(a)

我希望它能产生：

[[  1.   2.   3.   4.   5.   6.   7.   8.   9.  10.  11.]
 [ 12.  13.  14.  15.  16.  17.  18.  19.  20.  21.  22.]
 [ 23.  24.  25.  26.  27.  28.  29.  30.  31.  32.  33.]
 [ 34.  35.  36.  37.  38.  39.  40.  41.  42.  43.  44.]
 [ 45.  46.  47.  48.  49.  50.  51.  52.  53.  54.  55.]
 [ 56.  57.  58.  59.  60.  61.  62.  63.  64.  65.  66.]
 [ 67.  68.  69.  70.  71.  72.  73.  74.  75.  76.  77.]
etc for a 16 x 11 (16 rows and 11 column array)

有人可以通过明确的解释指出错误，并且也可能对更轻松的方法阐明了任何启示？(或替代方法(。我意识到我可能不需要第三个循环，但是正在努力弄清楚如何添加11的偏移。

以一种不太有效的方式进行此操作。...我发现这有效：

for i in range(11):
    a[0,i]=i+1
    a[1,i]=i+12
    a[2,i]=i+23
    a[3,i]=i+34
    a[4,i]=i+45
    a[5,i]=i+56
    a[6,i]=i+67
print(a)

添加了11的偏移，我不知道如何添加到我的现有结构中。

预先感谢您。

Answer 1

这有几种方法：

我一直使用的一个生成样品数组：

In [99]: np.arange(1,34).reshape(3,11)
Out[99]: 
array([[ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22],
       [23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33]])

考虑添加行和列均衡，这也很有用。Numpy Broadcasting使它变得容易：

In [100]: np.arange(0,33,11)[:,None]+np.arange(1,12)
Out[100]: 
array([[ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22],
       [23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33]])

等效列表代码：

In [101]: [[i+j for j in range(1,12)] for i in range(0,33,11)]
Out[101]: 
[[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11],
 [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22],
 [23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33]]

更接近您的尝试；i，j循环很好，但是您需要正确踩K：

In [106]: k=1
In [107]: out=np.zeros((3,11),int)
In [108]: for i in range(3):
     ...:     for j in range(11):
     ...:         out[i,j] = k
     ...:         k += 1
     ...:         
In [109]: out
Out[109]: 
array([[ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22],
       [23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33]])