使用split函数将数组arr横向分割成arr1和arr2
时间: 2023-08-08 12:06:31 浏览: 119
js中通过split函数分割字符串成数组小例子
可以使用NumPy库中的split函数将数组横向分割成两个新的数组。
假设我们有一个形状为 (m, n) 的数组 arr,我们希望将其分割成两个形状相同的数组 arr1 和 arr2。
可以使用以下代码实现:
```python
import numpy as np
arr1, arr2 = np.split(arr, 2, axis=1)
```
其中,split函数的第一个参数是要分割的数组,第二个参数是分割成几份,这里是分割成两份,第三个参数是指定分割的方向,这里是沿着水平方向(axis=1)进行分割。
执行完以上代码后,arr1 和 arr2 就是我们需要的两个横向分割后的数组。
阅读全文