首页 > 解决方案 > 使用 any 进行简单比较的更清洁方法?

问题描述

一些背景

我正在为用户提交的代码设置一些测试,以进行极其简单的数据科学测试。

编码“挑战”在 Jupyter Notepook 中的结构如下:

li1 = [2,4,3,-2,-1,5,10]
li2 = [5,4,-3,7,8,-5,10,9]
def findMedian(li):
    li.sort()

    # find the median of li...
    # your code below

    listlength = len(li)
    num = listlength//2
    if listlength % 2 == 0:
        return (li[num]+li[num-1])/2
    else:
        return li[num]
if findMedian(li1) == 3:
    print("1 is correct!")
else:
    print("1 is wrong")
if findMedian(li2) == 6:
    print("2 is correct!")
else:
    print("2 is wrong")

现在虽然这已经足够好了,但我决定我可能应该实现类似以下的东西来真正检查许多情况。

from statistics import median
import random

x = [list(range(random.randint(1,50))) for i in range(1,100)]

print(not any([False if findMedian(i) == median(i) else True for i in x]))

问题是,使用not any()感觉不对,我已经做了一段时间了,我觉得我的知识有点差距......

问题

有没有更简洁的方式来表达我的比较?

not any([False if findMedian(i) == median(i) else True for i in x]) 完成这项工作的同时,是否有更有效/pythonic 的方式来表达这一点?

标签: pythontesting

解决方案


首先,该列表是不必要的。改用生成器表达式,这将更有效,因为它允许any/all短路

接下来,False if x == y else True与 相同not x == y,对于包括整数在内的大多数对象x != y, 。

最后,由于德摩根定律,您可以使用all(...)代替not any(not ...).

all(findMedian(i) == median(i) for i in x)

推荐阅读