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]])]

相关用法

注：本文由纯净天空筛选整理自Jitender_1998大神的英文原创作品 Python | numpy.array_split() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。