修改Python3中包含集合的集合列表



我正在尝试创建一个以元组为元素的列表。每个元组有4个整数。前两个整数是压缩两个ranges的结果,而另两个则来自两个不同的整数。

我正在使用此代码创建元组和最终列表,该列表派生自笛卡尔乘积,如下所示:获取一系列列表的笛卡尔乘积?

import itertools
first_range = list(zip((10**exp for exp in range(0,7)),(10**exp for exp in range(1,8))))
second_range = list(zip((5*10**exp if exp != 1 else 10**2 for exp in range(1,8)),(5*10**exp for exp in range(2,9))))
final_list = list(itertools.product(first_range,second_range))

这个代码的问题是,最终结果看起来像这样:

[((1, 10), (100, 500)), ((1, 10), (500, 5000)), ((1, 10), (5000, 50000)), ((1, 10), (50000, 500000)), ((1, 10), (500000, 5000000)), ((1, 10), (5000000, 50000000)), ...

其中每个列表元素都是一个包含另外两个元组的元组,而我想要的是:

[(1, 10, 100, 500), (1, 10, 500, 5000), (1, 10, 5000, 50000), (1, 10, 50000, 500000), (1, 10, 500000, 5000000), (1, 10, 5000000, 50000000), ...

即,每个列表元素是包含4个整数的元组。

任何想法都将不胜感激。一定是在研究蟒蛇。编辑:由于ShadowRanger的评论,更新了代码的非工作部分

所以,当我发布这个问题时,我确信我已经接近答案了,但我没有意识到我已经如此接近了。解决额外元组问题的方法是:

import itertools
first_range = zip((10**exp for exp in range(7)),(10**exp for exp in range(1,8)))
second_range = zip((5*10**exp if exp != 1 else 10**2 for exp in range(1,8)),(5*10**exp for exp in range(2,9)))
iterator_of_tuples = itertools.product(first_range,second_range)
# the next line solves my issue
final_list = [x + y for x, y in iterator_of_tuples]

我所做的是一个简单的元组合并:如何在Python中合并两个元组。不知道为什么我没有早点想到

编辑:根据ShadowRanger的输入更新了答案

您的预期输出不是这两个范围的笛卡尔乘积。

如果你想要你的预期输出,这样的东西会起作用:

final_list = [(*x, *y) for x, y in zip(first_range, second_range)]

最新更新