numpy数组扩展的具体方式

Posted 2021-03-26

我有以下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，我的真实数组没有排序，并应最终保持未分类。

提前致谢！

答案

使用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])