我试图在函数中移动itertools.product
的用法。当我尝试这样做时,我会收到以下错误消息,我不知道为什么:
TypeError: 'int' object is not iterable
代码如下。在main
函数中,可以看到函数外部使用的算法,然后可以看到在函数中打包时使用的算法:
#!/usr/bin/env python
import itertools
def main():
elements_specification = [[10, 20], [30, 40], [50, 60]]
lists = [list(list_generated) for index, element_specification in enumerate(elements_specification) for list_generated in itertools.product(*elements_specification[:index + 1])]
for list_configuration in lists:
print(list_configuration)
print("---")
for list_configuration in list_element_combinations_variadic(
[[10, 20], [30, 40], [50, 60]]
):
print(list_configuration)
def list_element_combinations_variadic(
elements_specification
):
"""
This function accepts a specification of lists of elements for each place in
lists in the form of a list, the elements of which are lists of possible
elements and returns a list of lists corresponding to the combinations of
elements of the specification with varying numbers of elements.
For example, the list elements specification [[10, 20], [30, 40], [50, 60]]
yields the following lists:
[10]
[20]
[10, 30]
[10, 40]
[20, 30]
[20, 40]
[10, 30, 50]
[10, 30, 60]
[10, 40, 50]
[10, 40, 60]
[20, 30, 50]
[20, 30, 60]
[20, 40, 50]
[20, 40, 60]
"""
lists = [list(list_generated) for index, elements_specification in enumerate(elements_specification) for list_generated in itertools.product(*elements_specification[:index + 1])]
return lists
if __name__ == "__main__":
main()
发布于 2016-01-29 16:51:41
基本上,在main
方法和另一个方法之间有一个错误。
在main
中,element_specification
在for
中是正确的
for index, element_specification in enumerate(elements_specification)
但是在另一种方法中,elements_specification
在for
中
for index, elements_specification in enumerate(elements_specification)
它恰好是该方法的参数名称,所以您要在列表理解中重新分配该参数。
试一试这个
lists = [list(list_generated) for index, element in enumerate(elements_specification) for list_generated in itertools.product(*elements_specification[:index + 1])]
return lists
或者,由于您甚至不需要来自enumerate
的enumerate
,所以只需使用range
。
lists = [list(list_generated) for index in range(len(elements_specification)) for list_generated in itertools.product(*elements_specification[:index + 1])]
return lists
https://stackoverflow.com/questions/35089455
复制相似问题