Numpy 如何在两个轴上重复数组元素

Numpy是Python中用于数学运算和数组处理的最常用库之一。重复数组元素是数组处理中经常遇到的问题之一。本文将介绍如何使用Numpy在两个轴上重复数组元素。

阅读更多：Numpy 教程

基本的重复方法

Numpy中有几个函数可以重复数组元素，其中最基本的函数是repeat()函数。使用该函数的方法如下：

import numpy as np

a = np.array([1, 2, 3])
result = np.repeat(a, 3)

print(result)

运行该代码，输出结果为：

[1 1 1 2 2 2 3 3 3]

这个结果是将数组a中的每个元素重复3次。repeat()函数的第二个参数指定每个元素重复的次数。

如果想要在二维数组中重复元素，可以将数组转换为二维数组，然后再进行重复操作：

import numpy as np

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

print(result)

运行该代码，输出结果为：

[1 1 1 2 2 2 3 3 3 4 4 4 1 1 1 2 2 2 3 3 3 4 4 4]

这个结果是将数组a中的每个元素在列方向上重复3次。

在两个轴上重复元素

以上的方法只能在一个轴上重复元素。如果想要在两个轴上重复元素，可以使用tile()函数。使用该函数的方法如下：

import numpy as np

a = np.array([1, 2, 3])
result = np.tile(a, (2, 3))

print(result)

运行该代码，输出结果为：

[[1 2 3 1 2 3 1 2 3]
 [1 2 3 1 2 3 1 2 3]]

这个结果是将数组a中的每个元素在行方向上重复2次，在列方向上重复3次。

如果想要在二维数组中重复元素，可以将数组转换为三维数组，然后再进行重复操作：

import numpy as np

a = np.array([[1, 2], [3, 4]])
result = np.tile(a, (2, 3))

print(result)

运行该代码，输出结果为：

[[[1 2 1 2 1 2]
  [3 4 3 4 3 4]
  [1 2 1 2 1 2]
  [3 4 3 4 3 4]]

 [[1 2 1 2 1 2]
  [3 4 3 4 3 4]
  [1 2 1 2 1 2]
  [3 4 3 4 3 4]]]

这个结果是将数组a中的每个元素在行方向上重复2次，在列方向上重复3次，并且转换为三维数组后，深度方向上重复了2次。

总结

以上就是Numpy在两个轴上重复数组元素的方法。使用repeat()函数可以在一个轴上重复元素，使用tile()函数可以在两个轴上重复元素。重复元素是数组处理中非常常见的操作，掌握Numpy中的重复函数可以提高处理数组的效率。