Python新手疑问：combinations函数中的r参数来源解析

先拆解这段代码的核心逻辑：它是用来生成一个可迭代对象的幂集（也就是所有可能的子集，包括空集和全集）。

你困惑的r，来自代码里的range(len(s)+1)循环：

• 首先s = list(iterable)把输入转成列表，比如输入[1,2,3]时，len(s)是3，range(len(s)+1)等价于range(4)，会生成0,1,2,3四个整数。
• 生成器表达式combinations(s, r) for r in range(len(s)+1)会依次把这四个数赋值给r，再调用combinations：
  • 当r=0时：combinations(s, 0)生成包含空元组的迭代器（从集合里选0个元素只有空集一种可能）
  • 当r=1时：生成所有单个元素的组合，即(1,),(2,),(3,)
  • 当r=2时：生成所有两个元素的不重复组合，即(1,2),(1,3),(2,3)
  • 当r=3时：生成包含全部三个元素的组合(1,2,3)
• 最后chain.from_iterable把这些不同r对应的组合迭代器合并成一个连续迭代器，就得到了完整的幂集。

实际调用powerset([1,2,3])时，会依次输出：() (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)，和函数注释的结果完全一致。

内容的提问来源于stack exchange，提问作者Furqan_25