np.delete函数中axis的作用
时间: 2024-05-14 19:01:28 浏览: 72
webservice3 使用services.xml发布webservice
`np.delete` 函数可以从一个数组中删除指定的行或列。其中,`axis` 参数用于指定要删除的轴,它的作用如下:
- 当 `axis=None` 时,函数将展开整个数组并删除其中的指定元素。
- 当 `axis=0` 时,函数将删除指定的行。
- 当 `axis=1` 时,函数将删除指定的列。
例如,假设有一个二维数组 `arr`,其形状为 `(3, 4)`,我们想要删除第一行和第三行,可以使用以下代码:
``` python
import numpy as np
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
new_arr = np.delete(arr, [0, 2], axis=0)
```
其中,`[0, 2]` 表示要删除的行的索引,`axis=0` 表示按行删除。执行完上述代码后,`new_arr` 的值为:
```
array([[ 5, 6, 7, 8],
[ 1, 2, 3, 4]])
```
注意,`np.delete` 函数并不会修改原始数组 `arr`,而是返回一个新的数组 `new_arr`。
阅读全文