numpy数组扩展的具体方式

问题描述 投票:0回答:1

我有以下numpy数组(作为示例):

[0,1,5,4,3]

有没有办法“重复”这个数组,但以一种特定的方式:我需要一个形状25的最终数组,但排列方式,我会有5个零,然后 - 5个,然后 - 5个五,等等。期望输出的示例:

[0,0,0,0,0,1,1,1,1,1,
 5,5,5,5,5,4,4,4,4,4
 3,3,3,3,3]

如果我做np.append(arr, arr)-它将给我两个初始排序的数组,这不是我的目标BTW,我的真实数组没有排序,并应最终保持未分类。

提前致谢!

python arrays numpy
1个回答
1
投票

使用np.repeat

>>> np.repeat([0,1,5,4,3], 5)
array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 5, 5, 5, 5, 5, 4, 4, 4, 4, 4, 3, 3,
       3, 3, 3])
© www.soinside.com 2019 - 2024. All rights reserved.