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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。