与循环一起使用的 itertools 组合

Question

我有以下 Python 代码。因为使用的是随机数，所以每次都会生成一个新的答案：

import random

import numpy as np

N = 64 # Given

T = 5 # Given

FinalLengths = []

for i in range(T):

    c = range(1, N)
    x = random.sample(c, 2) # Choose 2 random numbers between 1 and N-1
    LrgstNode = max(x)
    SmlstNode = min(x)
    RopeLengths = [SmlstNode, LrgstNode - SmlstNode, N - LrgstNode] 
    S = max(RopeLengths)
    N = S
    FinalLengths.append(S)

avgS = np.mean(FinalLengths) # Find average

print("The mean of S is {}".format(avgS))

我的研究使我可能使用 itertools 组合来生成范围内的所有可能组合并让平均值收敛。如果有，怎么做？

谢谢。

Answer 1

它听起来就像你在追求这样的东西：

import random
import numpy as np
from itertools import combinations

N = 64 # Given
T = 5 # Given

FinalLengths = []

for i in range(T):
  c = list(range(1, N))
  for x in combinations(c, 2):
    S = max([min(x), max(x) - min(x), N - max(x)])
    N = S
    FinalLengths.append(S)

avgS = np.mean(FinalLengths) # Find average

print("The mean of S is {}".format(avgS))

要使用combinations(l, size)，我们可以传入一个列表l和每个组合元组的大小，以及int size。仅此而已！