np.repeat()

狼啸风云

修改于 2022-09-02 19:55:03

1.9K0

修改于 2022-09-02 19:55:03

文章被收录于专栏：计算机视觉理论及其实现

np.repeat()

np.repeat()用于将numpy数组重复。

numpy.repeat(a, repeats, axis=None);

参数：

axis=0,沿着y轴复制，实际上增加了行数 axis=1,沿着x轴复制，实际上增加了列数

1. 一维数组重复3次

# 随机生成[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次

# 二维数组在第一维和第二维分别重复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]]

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/02/07 ，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法