在一行中打印for循环结果并排序

Question

for n in range(1,1000000):
    print(n)
    result = []
    for x in range(1,3000001):
        if n%2==0:
            x=n/2
        else:
            x=3*n+ 1
        n=x
        result.append(n)
        if n==1:
            break
    print(len(result))
n+=1

我希望将这些结果打印在一个数组或类似的东西中。我的意思是这样。

3,1,7,2,5,8,1,..

那我想取最高的元素和它的索引。我该怎么做呢？谢谢。

Answer 1

第一个任务可以使用str.join：

>>> result = [4, 0, 9, 7, 8, 3, 2, 6, 1, 5]
>>> print (', '.join(map(str, result)))
4, 0, 9, 7, 8, 3, 2, 6, 1, 5

max 和 enumerate 用于第二个任务：

>>> ind, val = max(enumerate(result), key=lambda x:x[1])
>>> ind, val
(2, 9)

Answer 2

如果你将完成工作的循环分离到它自己的函数中，这会变得容易得多。

def collatz_length(n):
    result = []
    for x in range(1,3000001):
        if n%2==0:
            x=n/2
        else:
            x=3*n+ 1
        n=x
        result.append(n)
        if n==1:
            break
    return len(result)


print(max((collatz_length(i + 1), i) for i in range(1000000)))

由于你没有使用result，只是它的长度，你可以通过简单的计数来简化（并加速）函数

你可以使用ternary expression来整理x的计算

def collatz_length(n):
    for c in range(1, 3000001):
        x = 3 * n + 1 if n % 2 else n / 2
        n = x
        if n == 1:
            return c