从数学函数自动生成列表?

时间:2017-10-24 15:49:50

标签: python list python-3.6 collatz

我的想法是在任意范围内对以1,3,7和9结尾的数字运行3n + 1进程(Collatz conjecture),并告诉代码发送每个动作的长度到列表,所以我可以单独在该列表上运行函数。

到目前为止,我将单位数字1,3,7和9指定为:if n % 10 == 1; if n % 10 == 3 ...等等,我认为我的计划需要某种形式的嵌套for循环;我在列表追加时的位置是temp = []leng = [],并找到一种方法让代码在每次输入temp.clear()之前自动leng。我假设有不同的方法来做到这一点,我对任何想法持开放态度。

leng = []
temp = []
def col(n):
    while n != 1:
        print(n)
        temp.append(n)
        if n % 2 == 0:
            n = n // 2
        else:
            n = n * 3 + 1
    temp.append(n)
    print(n)

1 个答案:

答案 0 :(得分:0)

目前还不清楚你具体询问和想知道什么,所以这只是猜测。由于您只想知道序列的长度,因此无需在每个序列中实际保存数字 - 这意味着只创建了一个列表。

def collatz(n):
    """ Return length of Collatz sequence beginning with positive integer "n".
    """
    count = 0
    while n != 1:
        n = n // 2 if n % 2 == 0 else n*3 + 1
        count += 1
    return count

def process_range(start, stop):
    """ Return list of results of calling the collatz function to the all the
        numbers in the closed interval [start...stop] that end with a digit
        in the set {1, 3, 7, or 9}.
    """
    return [collatz(n) for n in range(start, stop+1) if n % 10 in {1, 3, 7, 9}]

print(process_range(1, 42))

输出:

[0, 7, 16, 19, 14, 9, 12, 20, 7, 15, 111, 18, 106, 26, 21, 34, 109]