跳转至

Python | numpy.array_split()方法

原文:https://www . geesforgeks . org/python-numpy-array _ split-method/

借助**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)

输出:

[数组([0,1,2]),数组([3,4]),数组([5,6]),数组([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)

输出:

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



回到顶部