Python numpy.array_split()方法

在numpy.array_split()方法的帮助下，我们可以通过使用numpy.array_split()方法得到具有不同维度的分割数组。

语法 : numpy.array_split()

返回：返回分割后的一维数组。

例子#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]])]