如何初始化Numpy数组的numpy数组

Question

我有一个尺寸为4x4的numpy array D

我想要一个基于用户定义值v的新numpy数组

如果v=2 ，则新的numpy数组应为[DD] 。 如果v=3 ，则新的numpy数组应为[DDD]

我如何将这样的numpy数组初始化为numpy.zeros(v)不允许我将数组作为元素放置？

Answer 1

如果我理解正确，您想获取一个2D数组并将其在第一个维度上平铺v次吗？ 您可以使用np.repeat ：

# a 2D array
D = np.arange(4).reshape(2, 2)

print D
# [[0 1]
#  [2 3]]

# tile it 3 times in the first dimension
x = np.repeat(D[None, :], 3, axis=0)

print x.shape
# (3, 2, 2)

print x
# [[[0 1]
#   [2 3]]

#  [[0 1]
#   [2 3]]

#  [[0 1]
#   [2 3]]]

如果希望输出保持二维，即(6, 2) 6，2 (6, 2) ，则可以省略[None, :]索引（有关numpy广播规则的更多信息，请参见此页 ）。

print np.repeat(D, 3, axis=0)
# [[0 1]
#  [0 1]
#  [0 1]
#  [2 3]
#  [2 3]
#  [2 3]]

另一个选择是np.tile ，其行为略有不同，因为它将始终平铺在最后一个维度上：

print np.tile(D, 3)
# [[0, 1, 0, 1, 0, 1],
#  [2, 3, 2, 3, 2, 3]])

Answer 2

您可以按照以下步骤进行操作：

import numpy as np
v = 3
x = np.array([np.zeros((4,4)) for _ in range(v)])

>>> print x
[[[ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]]

 [[ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]]

 [[ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]
  [ 0.  0.  0.  0.]]]

Answer 3

在这里，看看是否适合您。

import numpy as np
v = raw_input('Enter: ')

从用户输入初始化数组的numpy数组（显然可以是您想要的任何形状）：

b = np.zeros(shape=(int(v),int(v)))

我知道这不是在初始化一个numpy数组，但是由于您提到例如要在v为2的情况下想要[DD]数组，所以以为我也将其作为另一个选择。

new_array = []
for x in range(0, int(v)):
    new_array.append(D)