Python如何拆分numpy数组?

1 个回答|297 次阅读
Coco老师 - 公众号:量位学堂Qbit编程
擅长:编程教育

可以用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])]