检查二维numpy数组的每一行,如果重复,删去重复行
时间: 2023-12-22 19:06:41 浏览: 61
python 去除二维数组/二维列表中的重复行方法
可以使用numpy库中的unique函数来实现。具体步骤如下:
1. 使用unique函数对原始数组进行去重操作,设置参数axis=0表示按行去重。
2. 使用keep参数保留唯一行的索引,再使用索引操作从原始数组中筛选出唯一行。
下面是一个示例代码:
```python
import numpy as np
# 原始数组
arr = np.array([[1, 2, 3], [4, 5, 6], [1, 2, 3], [7, 8, 9], [4, 5, 6]])
# 按行去重
unique_arr, indices = np.unique(arr, axis=0, return_index=True)
# 筛选出唯一行
unique_rows = arr[np.sort(indices)]
print(unique_rows)
```
输出结果为:
```
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
```
可以看到,重复的两行被删除了,只保留了唯一行。
阅读全文