numpy中,二维数组A和二维数组B行数不同,列数相同。输出B中不存在于A中的行,并将这些行组成与B列数相同的二维数组
时间: 2023-02-08 20:20:16 浏览: 73
判断一维数组是否存在二维数组里 不存在就push到二维数组中形成二维数组,存在则不用管
可以使用numpy的 np.setdiff1d() 函数来实现。
``` python
import numpy as np
A = np.array([[1, 2], [3, 4], [5, 6]])
B = np.array([[1, 2], [3, 4], [7, 8], [9, 10]])
diff_rows = np.setdiff1d(B, A, axis=0)
diff_rows_arr = diff_rows.reshape(-1, B.shape[1])
```
diff_rows_arr 就是 B 中不存在于 A 中的行组成的二维数组。
例如:
```
diff_rows_arr
[[7 8]
[9 10]]
```
阅读全文