判断两个数组元素是否一致,一致返回1,不同返回0,返回数组。
第一种方法,遍历所有元素。代码如下:
def whethersame(a,b):
res =[]
for i in range(len(a)):
if a[i]==b[i]:
res.append(1)
else:
res.append(0)
return res
a=[1,2,3,4]
b=[1,3,4,2]
res=whethersame(a,b)
print(res)
输出结果为 :[1, 0, 0, 0]
第二种方法,一步到位
#判断数组是否一致
a = [1,2,3,4]
b = [1,3,4,2]
c = [int(aa==bb) for aa,bb in zip(a,b)]
print(c)
输出结果为:[1, 0, 0, 0]
感谢群友的帮助啦,第二种简直太方便了吧!