Python| numpy.array_split() 方法

借助numpy.array_split()方法，我们可以使用numpy.array_split()方法得到不同维度的分割数组。

Syntax : numpy.array_split()

Return : Return the splitted array of one dimension.

示例 #1：
在这个例子中，我们可以看到通过使用numpy.array_split()方法，我们可以通过将数组作为参数传递来将数组拆分为子数组的数量。

# import numpy
import numpy as np
  
array = np.arange(9)
  
# using numpy.array_split() method
gfg = np.array_split(array, 4)
  
print(gfg)

输出 ：

[array([0, 1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]

示例 #2：

# import numpy
import numpy as np
  
array = [[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]]
  
# using numpy.array_split() method
gfg = np.array_split(array, 3)
  
print(gfg)

输出 ：

[array([[1, 2, 3]]), array([[4, 5, 6]]), array([[7, 8, 9]])]