NumPy 中的集合运算举例: 查找共同元素,差异元素和共有元素

import numpy as np
# 创建一维 ndarray x
x = np.array([1,2,3,4,5])
# 创建一维 ndarray y
y = np.array([6,5,4,8,7])
# 输出 x
print('x = ', x)
# 输出 y
print('y = ', y)
# 比较x和y
print('使用intersect1d输出在x和y中共有的元素:', np.intersect1d(x,y))
print('使用setdiff1d输出在x中不在y中的元素:', np.setdiff1d(x,y))
print('使用union1d输出在x和y共有的元素:',np.union1d(x,y))

输出为:

x =  [1 2 3 4 5]
y =  [6 5 4 8 7]
使用intersect1d输出在x和y中共有的元素: [4 5]
使用setdiff1d输出在x中不在y中的元素: [1 2 3]
使用union1d输出在x和y共有的元素: [1 2 3 4 5 6 7 8]

猜你喜欢

转载自blog.csdn.net/Tyro_java/article/details/81360322