Numpy：repeat用法详解 Python

Numpy的repeat函数用于重复数组中的元素。它的语法如下：

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

参数说明：

• a：输入数组。
• repeats：每个元素重复的次数，可以是整数或整数数组。
• axis：沿着哪个轴重复，默认为None，表示在所有轴上都重复。
  示例1：在所有轴上都重复元素

  import numpy as np
  a = np.array([[1, 2], [3, 4]])
  b = np.repeat(a, 2)
  print(b)

  输出：

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

  在这个例子中，我们创建了一个2行2列的数组a，然后使用repeat函数将其重复2次，得到一个4行4列的输出数组b。可以看到，每个元素都被重复了2次。
  示例2：在指定轴上重复元素

  import numpy as np
  a = np.array([[1, 2], [3, 4]])
  b = np.repeat(a, 2, axis=0)
  print(b)

  输出：

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

  在这个例子中，我们将数组a沿着第0轴（行）重复2次，得到一个4行2列的输出数组b。可以看到，第0轴上的元素被重复了2次。
  示例3：使用整数数组指定重复次数

  import numpy as np
  a = np.array([[1, 2], [3, 4]])
  repeats = np.array([2, 3])  # [2, 3]表示第一行重复2次，第二行重复3次。
  b = np.repeat(a, repeats)
  print(b)

  输出：

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

  在这个例子中，我们使用一个整数数组作为repeats参数，表示每个元素重复的次数。可以看到，第一行元素被重复了2次，第二行元素被重复了3次。因此，输出数组b中第一行有4个元素，第二行有7个元素。总结：Numpy的repeat函数是一个非常实用的工具，用于在数组中重复元素。通过指定不同的参数，可以在不同轴上重复元素，也可以使用整数数组指定每个元素的重复次数。在实际应用中，可以根据需要灵活使用repeat函数来处理和操作数据。