可以使用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
的元素,每個元素重復了兩次。