python:如何使用相邻的重复列扩展numpy数组?

时间:2015-08-05 14:58:57

标签: python arrays numpy matrix

我想要实现的是我有一个像这样的矩阵:

axy
axy
axy
axy

我希望扩展该矩阵,使其成为:

aaaxxxyyy
aaaxxxyyy
aaaxxxyyy
aaaxxxyyy

是否有可以轻松管理此转换的功能?我想使用一种更好的方法,而不是分别平铺每一列并追加它们。

提前致谢。

1 个答案:

答案 0 :(得分:4)

您可以使用np.repeat

>>> import numpy as np
>>> a = np.arange(9).reshape(3,3)
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> b = np.repeat(a, 3, axis=1) # array, times, axis
>>> b
array([[0, 0, 0, 1, 1, 1, 2, 2, 2],
       [3, 3, 3, 4, 4, 4, 5, 5, 5],
       [6, 6, 6, 7, 7, 7, 8, 8, 8]])