python数组每N个取一个小数组
时间: 2023-09-08 19:16:44 浏览: 78
在Python中等距取出一个数组其中n个数的实现方式
你可以使用 Python 中的切片操作来实现这个功能。具体地,你可以先将原始数组按照每 N 个元素分割成若干个子数组,然后再将每个子数组提取出来。
以下是一个实现的示例代码:
```python
def split_array(arr, n):
return [arr[i:i+n] for i in range(0, len(arr), n)]
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
n = 3
sub_arrays = split_array(arr, n)
print(sub_arrays) # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]
```
在上面的代码中,`split_array` 函数接受两个参数:原始数组 `arr` 和每个子数组的长度 `n`。函数使用列表推导式将原始数组分割成若干个长度为 `n` 的子数组,并将它们组成一个新的列表返回。
阅读全文