numpy去除矩阵中重复的行
时间: 2023-05-26 17:02:23 浏览: 156
可以使用numpy.unique函数来去除矩阵中重复的行,设置参数axis=0即可按行去重。具体代码如下:
```python
import numpy as np
# 定义矩阵
mat = np.array([[1, 2, 3],
[4, 5, 6],
[1, 2, 3],
[7, 8, 9],
[4, 5, 6]])
# 对矩阵按行去重
unique_mat = np.unique(mat, axis=0)
print("原矩阵:\n", mat)
print("去重后矩阵:\n", unique_mat)
```
运行结果如下:
```
原矩阵:
[[1 2 3]
[4 5 6]
[1 2 3]
[7 8 9]
[4 5 6]]
去重后矩阵:
[[1 2 3]
[4 5 6]
[7 8 9]]
```
阅读全文