np.repeat 提供的复制一维数组的方法,是每个元素直接连续复制 n 次,而不是整体复制 n 次,下面实现了一种整体复制 n 次的方法。
a = np.arange(1, 5)
print(np.repeat(a, 3))
# 结果: [1 1 1 2 2 2 3 3 3 4 4 4]
print(a.reshape((1, a.size)).repeat(3, axis=0).reshape((a.size * 3,)))
# 结果: [1 2 3 4 1 2 3 4 1 2 3 4]
微信扫一扫
np.repeat 提供的复制一维数组的方法,是每个元素直接连续复制 n 次,而不是整体复制 n 次,下面实现了一种整体复制 n 次的方法。
a = np.arange(1, 5)
print(np.repeat(a, 3))
# 结果: [1 1 1 2 2 2 3 3 3 4 4 4]
print(a.reshape((1, a.size)).repeat(3, axis=0).reshape((a.size * 3,)))
# 结果: [1 2 3 4 1 2 3 4 1 2 3 4]
相关推荐