python - 列表理解而不是循环扩展

Question

我可以将这段代码写成一行吗？我尝试在列表理解中使用链。

def divisors(n):
    result = []
    for div in range(1, int(sqrt(n)) + 1):
        if n % div == 0:
            result.extend([div, n / div])
    return list(set(result))

Answer 1

一套理解似乎是合适的。另请注意，我使用的是 // 而不是 /，因为 float 与此问题无关。

from math import sqrt

def divisors(n):
    return {x for div in range(1, int(sqrt(n)) + 1) 
              if not (n % div)
              for x in [div, n // div]}

divisors(15)
# {1, 3, 5, 15}

如果你真的想要一个列表，把集合变成一个列表很容易。

list(divisors(15))
# [1, 3, 5, 15]