如何将这段代码的结果转换为列表或集合？

Question

如何将这段代码的结果转换为列表或集合？

def factors(x):
    for i in range(1, x + 1):
        if x % i == 0:
            print(i)

num = 100
factors(num)

Answer 1

您需要退货而不是打印。 您可以在函数中建立一个列表：

def factors(x):
    result = []
    for i in range(1, x + 1):
        if x % i == 0:
            result.append(i)
    return result

num = 100
print(factors(num))

如果需要，也可以使用result = set()和result.add()将其设置为一个集合。

另一个选择是将函数转换为生成器 ； 您必须在调用函数后进行一些迭代：

def factors(x):
    for i in range(1, x + 1):
        if x % i == 0:
            yield i

num = 100
for factor in factors(num):
    print(factor)

但您也可以将所有值“拉入”列表或设置中：

factors_list = list(factors(num))
factors_set = set(factors(num))

Answer 2

检查除数直到值的平方根。 如果匹配，则将除数和商添加到因子列表中。 然后从列表中创建一个集合，以删除重复的元素（仅当x是整数的平方时才会出现）。 排序后最后返回列表，以使返回的列表升序排列。

import math
def factors(x):
    f = []
    for i in range(1, math.ceil(math.sqrt(x))+1):
        if x % i == 0:
            f.append(i)
            f.append(x/i)
    return sorted(set(f))