python - Numpy，以不同顺序复制数据

Question

例如我有数据

>>> a = np.array([1, 2, 3, 4])
>>> b = np.array([3, 4, 5, 6])

我想将每个向量中的每个项目复制到向量长度的值。所以结果可以是

>>> a2 = np.array([1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4])
>>> b2 = np.array([3, 4, 5, 6, 3, 4, 5, 6, 3, 4, 5, 6, 3, 4, 5, 6])

使用np.tile(b, len(b))可以输出b2。但是，我怎样才能得到 a2？

Answer 1

这两个复制有点不同。第一个可以用.repeat(..) [numpy-doc]获得:

>>> a.repeat(len(a))
array([1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4])

第二个 .tile(..) [numpy-doc] :

>>> np.tile(b, len(b))
array([3, 4, 5, 6, 3, 4, 5, 6, 3, 4, 5, 6, 3, 4, 5, 6])