如何去掉二维array的某个元素
时间: 2023-07-05 13:29:21 浏览: 84
可以使用numpy.delete方法来删除二维数组中的某个元素。delete方法的语法如下:
```python
numpy.delete(arr, obj, axis=None)
```
其中,arr表示要删除元素的数组;obj是要删除的元素的下标或者下标组成的列表;axis参数可选,默认值为None,表示要删除的元素所在的轴。
例如,删除一个二维数组中的第一行可以使用如下代码:
```python
import numpy as np
arr2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr2d = np.delete(arr2d, 0, axis=0)
print(arr2d)
```
输出结果为:
```
[[4 5 6]
[7 8 9]]
```
如果要删除多个元素,可以将它们的下标组成一个列表。
例如,删除一个二维数组中的第一行和第三行可以使用如下代码:
```python
import numpy as np
arr2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr2d = np.delete(arr2d, [0, 2], axis=0)
print(arr2d)
```
输出结果为:
```
[[4 5 6]]
```
如果要删除某一列,可以将axis参数设置为1。
例如,删除一个二维数组中的第二列可以使用如下代码:
```python
import numpy as np
arr2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr2d = np.delete(arr2d, 1, axis=1)
print(arr2d)
```
输出结果为:
```
[[1 3]
[4 6]
[7 9]]
```
阅读全文