Python 快速扩充一维数组到指定元素个数

今天仿真的时候遇到了一个问题，就是如何将一个一维数组扩充到想要的元素个数。比如：

一维数组中只有单个元素时

from numpy import array, repeat

x = array([1])
num = 10
y = repeat(x, repeats=num)
print("原始数组为：")
print(x)
print("扩充后的数组为：")
print(y)
"""
原始数组为：
[1]
扩充后的数组为：
[1 1 1 1 1 1 1 1 1 1]
"""

一维数组中存在多个元素时

from numpy import array, repeat

x = array([1, 1])
num = 10
y = repeat(x[0], repeats=num)
print("原始数组为：")
print(x)
print("扩充后的数组为：")
print(y)
"""
result:
原始数组为：
[1]
扩充后的数组为：
[1 1 1 1 1 1 1 1 1 1]
"""

如果大家觉得有用，就点个赞让更多的人看到吧~