Python中如何用通过索引的方式来删除,使得arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])变成arr = np.array([[1, 2, 3], [7, 8, 9]])
时间: 2024-09-28 14:05:58 浏览: 33
Python 获取numpy.array索引值的实例
在Python的NumPy库中,如果你想要通过索引来删除二维数组的一部分元素,可以直接使用`numpy.delete()`函数配合数组切片。但是这个函数主要是用来删除一维数组中的元素,对于二维数组的操作通常需要两步。首先,你可以使用`numpy.delete()`删除一维数组的对应部分,然后利用切片更新二维数组的指定位置。
假设你想从给定的数组`arr`中删除第二行,可以按照下面的步骤操作:
```python
import numpy as np
# 初始化二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 删除指定行
row_to_remove = 1 # 行索引从0开始,所以这里是删除第二行
arr = np.delete(arr, row_to_remove, axis=0) # axis=0表示按行删除
# 结果变成了 arr = [[1, 2, 3], [7, 8, 9]]
print(arr)
# 或者,如果你想要保持原始数组不变,可以复制后再操作
# arr_copy = arr.copy() # 创建数组副本
# arr_copy = np.delete(arr_copy, row_to_remove, axis=0)
# print(arr_copy)
```
阅读全文