10

我正在研究Project Euler #35，我需要找到一个数字的循环排列。使用itertools，我可以很容易地得到一个数字的排列。但是，我想用列表推导来做到这一点（因为它看起来更像 Pythonic；我也在尝试熟悉列表推导）。

我发现所有的圆形素数只能包含数字 1、3、7 和 9（这不包括 2 和 5，它们根据定义是圆形素数）。如果任何其他数字在数字（0、2、4、5、6 或 8）中，则其中一个排列将不是质数（因为该数字将在至少一个排列中的最后一个）。

因此，我尝试这样做：

from itertools import permutations
l = [x for x in list(permutations('1397', y)) for y in range(7)]

我需要使用y for y in range(7)，以便获得不同长度的排列。

然而，这给了我一个TypeError：

Traceback (most recent call last):
  File "<pyshell#23>", line 1, in <module>
    l = [x for x in list(permutations('1397', y)) for y in range(7)]
TypeError: an integer is required

这可行，但它没有在一个列表理解中使用两个变量：

l = []
for y in range(7):
    l.append([x for x in list(permutations('1379', y))])

如何进行双变量列表理解？谢谢！

4

2 回答 2

14

该for y in range(7)部分应位于排列循环之前。：

l = [x for y in range(7) for x in list(permutations('1397', y))]

上面的列表理解等价于：

In [93]: l = []

In [94]: for y in range(7):
    ...:     l.extend(list(permutations('1397', y)))

例如：

In [76]: l = [x for y in range(3) for x in list(permutations('1397', y))]

In [77]: l
Out[77]: 
[(),
 ('1',),
 ('3',),
 ('9',),
 ('7',),
 ('1', '3'),
 ('1', '9'),
 ('1', '7'),
 ('3', '1'),
 ('3', '9'),
 ('3', '7'),
 ('9', '1'),
 ('9', '3'),
 ('9', '7'),
 ('7', '1'),
 ('7', '3'),
 ('7', '9')]

list-comprehension以及您的工作示例的版本，

l = []
for y in range(7):
    l.append(list(permutations('1397', y)))

是：

In [85]: l = [list(permutations('1397', y)) for y in range(3)]

In [86]: l
Out[86]: 
[[()],
 [('1',), ('3',), ('9',), ('7',)],
 [('1', '3'),
  ('1', '9'),
  ('1', '7'),
  ('3', '1'),
  ('3', '9'),
  ('3', '7'),
  ('9', '1'),
  ('9', '3'),
  ('9', '7'),
  ('7', '1'),
  ('7', '3'),
  ('7', '9')]]
于 2013-01-19T17:00:01.683 回答
2 
[list(permutations('1397',x)) for x in range(7)]
于 2013-01-19T17:01:35.497 回答