python - 断言所有值都是负数或所有值都是非负数

Question

这个问题已经有答案了:

Check if all numbers in a list are same sign in Python? (4 个回答)

已关闭 5 年前。

我需要断言数组中的所有数值要么是负数，要么是非负数。

我写的是:

def check(arr):
    return all([i < 0 for i in arr]) or all([i >= 0 for i in arr])

我认为这稍微更有效率:

def check(arr):
    temp = [i < 0 for i in arr]
    return all(temp) or not any(temp)

我想知道是否有更干净/更Pythonic的方式，或者也许是一些我可以使用的算术技巧。

Answer 1

一种方法是使用集合推导式来派生一组 bool 值。该集合将为{True}、{False}或{True, False}。然后测试您的 set 的长度是否等于 1。

def check(arr):
    return len({i < 0 for i in arr}) == 1