python中np.array删除元素
时间: 2023-09-25 13:16:30 浏览: 120
对Python 字典元素进行删除的方法
5星 · 资源好评率100%
在 NumPy 中,可以使用 `np.delete()` 函数来删除数组中的元素。具体的用法如下:
```python
import numpy as np
# 创建一个 NumPy 数组
arr = np.array([1, 2, 3, 4, 5])
# 删除第二个元素
new_arr = np.delete(arr, 1)
print(new_arr) # [1 3 4 5]
```
在这个例子中,我们首先创建了一个包含整数 1 到 5 的 NumPy 数组。然后,我们使用 `np.delete()` 函数删除了数组中的第二个元素(索引为 1)。最后,我们打印出了删除元素后的新数组。
需要注意的是,这个函数会返回一个新的数组,原始数组不会被修改。如果想在原始数组上进行修改,可以直接使用切片操作来删除元素。例如:
```python
import numpy as np
# 创建一个 NumPy 数组
arr = np.array([1, 2, 3, 4, 5])
# 删除第二个元素
arr = np.concatenate([arr[:1], arr[2:]])
print(arr) # [1 3 4 5]
```
在这个例子中,我们使用 `np.concatenate()` 函数将两个切片拼接成一个新的数组,并将其赋值给原始数组。这样做就可以在原始数组上进行修改了。
阅读全文