Numpy中如何重复arange数组

在本文中，我们将介绍如何使用Numpy重复一个arange数组。arange是一个用于创建一组连续数字的Numpy函数，重复arange数组意味着将该数组作为一个单元重复几次，返回一个更大的数组。

阅读更多：Numpy 教程

语法

在Numpy中，使用repeat函数进行数组重复，其语法如下：

numpy.repeat(arr, repetitions, axis=None)

• arr：要重复的数组。
• repetitions：指定从原始数组中重复元素的次数。
• axis：指定重复的轴方向，默认为None。当指定为None时，将原始数组展平，并重复其元素。

示例

下面我们将通过几个示例，介绍如何使用Numpy完成arange数组的重复。

示例1：简单的重复

在第一个示例中，我们将创建一个长度为3的arange数组并将其重复3次。代码如下：

import numpy as np

a = np.arange(3)
b = np.repeat(a, 3)

print(b)

输出结果如下：

array([0, 0, 0, 1, 1, 1, 2, 2, 2])

示例2：按轴方向重复

在第二个示例中，我们将创建一个二维数组，并按列方向重复该数组。代码如下：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.repeat(a, [2, 3], axis=1)

print(b)

输出结果如下：

array([[1, 1, 2, 2, 2],
       [3, 3, 4, 4, 4]])

示例3：使用tile函数

在第三个示例中，我们将使用Numpy中的tile函数重复arange数组。tile函数将数组沿指定的方向重复n次，代码如下：

import numpy as np

a = np.arange(3)
b = np.tile(a, 3)

print(b)

输出结果如下：

array([0, 1, 2, 0, 1, 2, 0, 1, 2])

总结

在本文中，我们介绍了如何使用Numpy重复arange数组。可以使用repeat函数按指定的方向重复数组，也可以使用tile函数按指定次数重复数组。Numpy的这些函数使数组处理变得更加便捷。