怎么从numpy数组中删除指定下标的数据
时间: 2024-05-07 13:16:03 浏览: 111
可以使用numpy.delete()函数来删除指定下标的数据。
语法:numpy.delete(arr, obj, axis=None)
参数:
- arr:要删除元素的数组。
- obj:要删除的元素的下标或切片。
- axis:指定要删除元素的轴。如果未提供,则默认为None,即展平输入数组,然后删除元素。
示例代码:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print("原始数组:", arr)
# 删除下标为2的元素
new_arr = np.delete(arr, 2)
print("删除下标为2的元素后的数组:", new_arr)
# 删除下标为1和3的元素
new_arr = np.delete(arr, [1, 3])
print("删除下标为1和3的元素后的数组:", new_arr)
# 删除二维数组中第一列
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 0, axis=1)
print("删除第一列后的二维数组:")
print(new_arr)
```
输出结果:
```
原始数组: [1 2 3 4 5]
删除下标为2的元素后的数组: [1 2 4 5]
删除下标为1和3的元素后的数组: [1 3 5]
删除第一列后的二维数组:
[[2 3]
[5 6]
[8 9]]
```
阅读全文