np.array([[[1, 3]], [[550, 92]], [[7, 8]]]),如何除去元素[[7, 8]],给出普遍的代码
时间: 2023-10-15 20:07:15 浏览: 48
numpy.arrayh函数简介(博客资源备份)
可以使用numpy中的delete函数来删除数组中的特定元素。代码如下:
```python
import numpy as np
arr = np.array([[[1, 3]], [[550, 92]], [[7, 8]]])
arr = np.delete(arr, 2, axis=0)
print(arr)
```
输出结果:
```
array([[[ 1, 3]],
[[550, 92]]])
```
其中的delete函数的参数含义如下:
- 第一个参数:需要删除元素的数组
- 第二个参数:需要删除的元素的索引,这里是2,即删除第三个元素
- 第三个参数:删除元素的维度,这里是0,即删除第一个维度上的元素。由于我们的数组是三维的,所以需要指定维度。
阅读全文