在NumPy中,可以使用numpy.split()
函數來分割數組。該函數接受三個參數,分別為待分割的數組、分割點或分割的位置、沿著哪個軸進行分割。
示例如下:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 將數組arr分割成兩部分
splitted_arr = np.split(arr, [3])
print(splitted_arr)
輸出結果為:
[array([1, 2, 3]), array([ 4, 5, 6, 7, 8, 9, 10])]
在上面的示例中,我們將數組arr
按照索引3進行分割,得到了兩個子數組。