np.repeat()
np.repeat()用于将numpy数组重复。
numpy.repeat(a, repeats, axis=None);
参数:
axis=0,沿着y轴复制,实际上增加了行数 axis=1,沿着x轴复制,实际上增加了列数
1. 一维数组重复3次
代码语言:javascript复制# 随机生成[0, 5)之间的数,形状1行4列,将此数组按y轴重复3次
import numpy as np
pop = np.random.randint(0, 5, size=(1, 4)).repeat(3, axis=0)
print(pop)
[[1 4 0 4]
[1 4 0 4]
[1 4 0 4]]
2. 二维数组在第一维和第二维分别重复3次
代码语言:javascript复制# 二维数组在第一维和第二维分别重复3次
pop_reshape = pop.reshape(2, 6)
pop_reshape_repeataxis0 = pop_reshape.repeat(3, axis=0)
pop_reshape_repeataxis1 = pop_reshape.repeat(3, axis=1)
print(pop_reshape)
print(pop_reshape_repeataxis0)
print(pop_reshape_repeataxis1 )
[[1 4 0 4 1 4]
[0 4 1 4 0 4]]
[[1 4 0 4 1 4]
[1 4 0 4 1 4]
[1 4 0 4 1 4]
[0 4 1 4 0 4]
[0 4 1 4 0 4]
[0 4 1 4 0 4]]
[[1 1 1 4 4 4 0 0 0 4 4 4 1 1 1 4 4 4]
[0 0 0 4 4 4 1 1 1 4 4 4 0 0 0 4 4 4]]