python itertools 组合可扩展性

Question

我设法修改了我在 Internet 上找到的示例代码，以从一个文件夹中找到一组文件的所有可能组合，每对 2 个。

如果我有一个包含以下文件的文件夹 test:file1、file2、file3、file4 并运行以下代码:

import os, itertools, glob
folder = "test"

files = glob.glob(folder + "/*")
counter = 0
for file1, file2 in itertools.combinations(files, 2):
  counter = counter + 1
  output = file1 + " and " + file2
  print output, counter

我的输出是这样的:

test/file1 and test/file2 1
test/file1 and test/file3 2
test/file1 and test/file4 3
test/file2 and test/file3 4
test/file2 and test/file4 5
test/file3 and test/file4 6

这非常适合列出所有可能的 2 个文件组而不重复。现在，由于我对 for 循环进行了硬编码，因此在将其扩展到“x”文件组的同时保持代码简单时遇到了问题。 IE，我希望用户选择“x”，这样如果他选择 3，脚本将显示输出:

test/file1 and test/file2 and test/file3 1
test/file1 and test/file2 and test/file4 2
test/file1 and test/file3 and test/file4 3
test/file2 and test/file3 and test/file4 4

整个想法并不是在标准输出上实际显示输出，而是将它们用作子进程调用中的参数。

有什么建议吗？

Answer 1

x=3

for combination in itertools.combinations(files, x):
  counter = counter + 1
  output = " and ".join(combination)
  print output, counter

可以使用 sys.argv 获取命令行参数