np.split示例
时间: 2023-06-21 09:22:59 浏览: 140
用Pipelined Table实现split函数的示例
下面是一个简单的np.split()函数的示例:
```python
import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
sub_arrays = np.split(arr, 2, axis=1)
print(sub_arrays)
```
输出结果为:
```
[array([[ 1, 2],
[ 5, 6],
[ 9, 10]]),
array([[ 3, 4],
[ 7, 8],
[11, 12]])]
```
上述示例中,我们使用np.array()函数创建了一个二维数组arr,并将其沿着axis=1的方向分割成了两个子数组。分割后的结果是一个列表sub_arrays,其中包含两个元素,每个元素都是一个分割后的子数组。
阅读全文