python 使用np.random模块随机生成两个数组，找出两个一维数组中相同的元素。

话不多说，先上代码

A = np.random.randint(0,10,(1,8))
B = np.random.randint(0,10,(1,8))
print(A)
print(B)  # 为了方便查看对比，把两个数组也输出
a = np.squeeze(A)  # 删除数组一维条目
b = np.squeeze(B)
set(a)&set(b)

运行结果

[[8 5 1 9 2 9 5 5]]
[[0 7 5 3 4 0 8 4]]
{5, 8}

使用集合set交集直接得出两个随机生成的数组，先要把一维数组的外面的条目去掉，如果没有去掉，直接使用set()强制转换是会报错的。