量化交易之Python篇 - 取两个数组的 交、并、差集合

data_list = ["小米", "小明", "小强", "小红", "小蓝"]
target_list = ["小米", "小明", "小强", "小白", "小黑"]


# 取 两个数组的交集
print(list(set(data_list).intersection(set(target_list))))  # ['小明', '小米', '小强']


# 取 两个数组的并集
print(list(set(data_list).union(set(target_list))))  # ['小蓝', '小米', '小红', '小白', '小强', '小明', '小黑']


# 取 两个数组的差集 (data_list 有, target_list 没有)
print(list(set(data_list).difference(set(target_list))))  # ['小蓝', '小红']


# 取 两个数组的差集 (target_list 有, data_list 没有)
print(list(set(target_list).difference(set(data_list))))  # ['小白', '小黑']