np.repeat用法

O_meGa

已于 2024-01-05 18:53:58 修改

阅读量1k

点赞数 7

分类专栏： numpy python 文章标签： python

于 2024-01-05 18:51:25 首次发布

本文链接：https://blog.csdn.net/sinat_34461199/article/details/135416128

版权

python 同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

numpy

18 篇文章 0 订阅

订阅专栏

本文详细介绍了NumPy库中的np.repeat函数，用于对数组元素进行重复，区分了不同情况下按行或按列重复以及与np.stack区别的讲解。

摘要由CSDN通过智能技术生成

np.repeat用于对数组中的元素进行重复。它具有以下语法格式：

np.repeat(a, repeats, axis=None)

其中，a是要重复的数组，repeats是指定每个元素重复的次数，而axis是指定重复的轴的参数。

一些示例：

import numpy as np

# 示例1：对一维数组进行元素重复
a = np.array([1, 2, 3])
b = np.repeat(a, repeats=3)
print(b)
# 输出：
# [1 1 1 2 2 2 3 3 3]

# 示例2：对二维数组按行进行元素重复
x = np.array([[1, 2], [3, 4]])
y = np.repeat(x, repeats=2, axis=0)
print(y)
# 输出：
# [[1 2]
#  [1 2]
#  [3 4]
#  [3 4]]

# 示例3：对二维数组按列进行元素重复
x = np.array([[1, 2], [3, 4]])
y = np.repeat(x, repeats=3, axis=1)
print(y)
# 输出：
# [[1 1 1 2 2 2]
#  [3 3 3 4 4 4]]

注意与np.stack的区别：np.stack会改变array的轴的数量，而np.repeat不会改变轴的数量。