在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()函数的用法,你可以根据自己的需求灵活使用该函数。