访问作为可选参数传递的 class 个实例的属性

Question

我有两个具有列表属性的 class 实例，例如：

foo.range = [1,2]
bar.range = [3,4]

我还有一个接受多个参数的函数：

def permutations(*args):
    return list(itertools.product(arg.range for arg in args))

我想要 permutations(foo, bar) 到 return 这两个（或更多）列表（即 [(1,3), (1,4) …]）的所有排列，但我实际上得到 [([1, 2],), ([3, 4],)]

有人可以帮助我了解我哪里出错了以及如何实现我希望的结果吗？

Answer 1

itertools.product 期望迭代作为单独的参数（itertools.product(*iterables[, repeat])），因此您需要使用 splat（*) 运算符：

>>> def permutations(*args):
        return list(itertools.product(*(arg.range for arg in args)))

>>> permutations(foo, bar)
[(1, 3), (1, 4), (2, 3), (2, 4)]

考虑这个简单的函数，它将在 args 中收集传递给它的所有位置参数：

>>> def func(*args):
    print args
    print list(args[0])
...

当我们向它传递一个生成器表达式时，它被收集为 args 元组中的一个单独项目：

>>> func(x for x in range(5))
(<generator object <genexpr> at 0x7fda2ab4e780>,)
[0, 1, 2, 3, 4]

为了解决这个问题，我们需要解压缩我们的生成器表达式，同时像 *x for x in range(5) 这样的东西在 Python 中不是有效语法，我们需要在生成器表达式周围添加额外的括号：

>>> func(*(x for x in range(5)))
(0, 1, 2, 3, 4)
# Failed as expected for args[0]
Traceback (most recent call last):
  File "<ipython-input-215-13af4340dad1>", line 1, in <module>
    func(*(x for x in range(5)))
  File "<ipython-input-213-08941e2a3c06>", line 3, in func
    print list(args[0])
TypeError: 'int' object is not iterable

访问作为可选参数传递的 class 个实例的属性

Accessing attributes of class instances passed as optional arguments

python

class

instance

argument-passing

python-2.7