Python如何拆分numpy数组？

可以用array_split()拆分数组

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
newarr = np.array_split(arr, 3)
print(newarr)

执行结果：

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