python之使用numpy求两个二维数组的差集

import numpy as np

a0 = np.asarray([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
a1 = np.asarray([[1, 2, 3], [4, 5, 6]])


def calArray2dDiff(array_0, array_1):
    array_0_rows = array_0.view([('', array_0.dtype)] * array_0.shape[1])
    array_1_rows = array_1.view([('', array_1.dtype)] * array_1.shape[1])

    return np.setdiff1d(array_0_rows, array_1_rows).view(array_0.dtype).reshape(-1, array_0.shape[1])


print(calArray2dDiff(a0, a1))

[[7 8 9]]