Python 判断列表中两个0、两个非0且包含一个1的实现方法
使用以下 Python 代码可以实现判断列表中是否包含两个 0、两个非 0 且其中包含一个 1:
def check_list(lst):
count_zero = 0
count_nonzero = 0
count_one = 0
for num in lst:
if num == 0:
count_zero += 1
elif num != 0 and num != 1:
count_nonzero += 1
elif num == 1:
count_one += 1
if count_zero == 2 and count_nonzero == 2 and count_one == 1:
return True
else:
return False
# 测试
lst1 = [0, 1, 2, 0, 3]
print(check_list(lst1)) # 输出 True
lst2 = [0, 1, 2, 0, 3, 4]
print(check_list(lst2)) # 输出 False
在这个实现中,我们使用了三个计数器来分别计算 0 的个数、非 0 的个数和 1 的个数。最后判断计数器的值是否满足条件,如果满足则返回 True,否则返回 False。
原文地址: https://www.cveoy.top/t/topic/jJkd 著作权归作者所有。请勿转载和采集!