python 离散数学 判断单射 双射 满射

# 离散数学 判断单射 双射 满射
def panduan(sets):# 单射标志danshe = True# 满射标志manshe = True# 将传入对象分隔开,存入两个变量set1, set2 = sets
# 判断是否单射# 遍历set1,set2中所有数据,是否在 set2,set1中存在映射,判断是否满足单射或者满射for i in set1:# 单射判断# 如果set2中存在两个及以上或不存在与set1中元素对应的映射,则不满足单射条件,跳出循环if set2.count(i)!=1:danshe = Falsebreakfor i in set2:# 满射判断# 如果set1中不存在一个及以上与set2中元素对应的映射,则不满足满射条件,跳出循环if set1.count(i) == 0:manshe = Falsebreakif danshe and manshe:print("双射")elif danshe:print("单射")elif manshe:print("满射")else:print("既不是单射也不是满射")if __name__ == '__main__':a = eval(input('请输入两个集合或者列表或者元组以","或" "隔开').replace(' ', ","))panduan(a)