NumPy怎么沿特定轴重复元素

numpy

423

2024/4/19 12:44:12

栏目: 编程语言

可以使用np.repeat()函数来沿特定轴重复元素。这个函数接受三个参数：要重复的数组、重复次数和轴的索引。以下是一个示例：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
repeated_arr = np.repeat(arr, 2, axis=0)  # 沿第0轴重复元素
print(repeated_arr)

这将输出：

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

在这个示例中，我们沿第0轴重复了原始数组arr的元素，每个元素重复了两次。

最新知识库