python 将数组按额分配,最后一个数组可以长度小一些,其他数组长度要一致
时间: 2024-01-26 17:05:07 浏览: 115
Python3实现将一维数组按标准长度分隔为二维数组
好的,我明白了。您可以使用 Python 中的 `numpy.array_split` 函数来实现这个需求。这个函数会将一个数组按照指定的均匀大小分割成多个子数组,最后一个子数组的长度可能会小于指定大小。
例如,假设您有一个长度为 10 的数组 `arr`,您希望将它分割成 3 个长度相等的子数组,可以这样实现:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
split_arr = np.array_split(arr, 3)
print(split_arr)
```
输出为:
```
[array([1, 2, 3, 4]), array([5, 6, 7]), array([8, 9, 10])]
```
可以看到,`split_arr` 中包含了 3 个长度不同的子数组,每个子数组的长度都尽可能相等。
阅读全文