NumPy中可以使用split
函數來拆分數組,語法如下:
numpy.split(ary, indices_or_sections, axis=0)
其中,ary
表示要拆分的數組,indices_or_sections
可以是一個整數或者一個數組,用于指定拆分的位置或者拆分的個數,axis
表示沿著哪個軸進行拆分。
例如,將一個一維數組拆分成多個子數組:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr, 3)
print(subarrays)
# [array([1, 2]), array([3, 4]), array([5, 6])]
將一個二維數組按行拆分成多個子數組:
arr = np.array([[1, 2], [3, 4], [5, 6]])
subarrays = np.split(arr, 3)
print(subarrays)
# [array([[1, 2]]), array([[3, 4]]), array([[5, 6]])]
需要注意的是,拆分后得到的子數組是視圖而不是副本,修改子數組會影響原始數組。