在NumPy中,split()函數用于將數組拆分為子數組。split()函數有三個參數:數組、拆分點和軸。
參數解釋:
下面是使用split()函數的幾個示例:
import numpy as np
# 示例1:將數組拆分成兩個子數組
arr1 = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr1, 2)
print(subarrays)
# 輸出:[array([1, 2, 3]), array([4, 5, 6])]
# 示例2:將數組拆分成三個子數組
arr2 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
subarrays = np.split(arr2, 3)
print(subarrays)
# 輸出:[array([[1, 2, 3]]), array([[4, 5, 6]]), array([[7, 8, 9]])]
# 示例3:在指定索引處拆分數組
arr3 = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr3, [2, 4])
print(subarrays)
# 輸出:[array([1, 2]), array([3, 4]), array([5, 6])]
以上示例展示了split()函數的用法,你可以根據自己的需求靈活使用該函數。