python：验证条件和引发异常的最佳方法

发布时间：2020-12-20 12:17:48 所属栏目：Python 来源：网络整理

导读：嗨我必须验证一个向量是否包含全0或1,如果没有引发异常： def assert_all_zero_or_one(vector): if set(vector)=={0}: return 0 if set(vector)=={1}: return 1 raise TypeError 有了这个优点 assert_all_zero_or_one([1,1,1]) # return 1assert_all_zero_o

嗨
我必须验证一个向量是否包含全0或1,如果没有引发异常：

def assert_all_zero_or_one(vector):
    if set(vector)=={0}: return 0
    if set(vector)=={1}: return 1
    raise TypeError

有了这个优点

assert_all_zero_or_one([1,1,1]) # return 1
assert_all_zero_or_one([0,0]) # return 0
assert_all_zero_or_one([1,1]) # raise TypeError

我不喜欢这个解决方案..用python有一个最好的方法吗？

解决方法

def allOneOf(items,ids):
    first = items[0]
    if first in ids and all(i==first for i in items):
        return first
    else:
        raise TypeError()

assert_all_zero_or_one = (lambda vector: allOneOf(vector,set([0,1])))

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!