Python,压扁一个丑陋的嵌套for循环

标签 python list dictionary python-itertools cartesian-product

我想把几个数据文件通过两个模块来处理它们,使用每个模块的几个参数的几个设置的每个组合。执行此操作的明显方法是使用嵌套 for 循环,但是当您达到 7+ 嵌套 for 循环时,没有。我想让这个比那个更优雅。

我已经读过几个非常相似的问题,但是 this one表明我可能想使用 itertools,它只遍历数字序列,而我想遍历作为值包含在字典中的字符串列表; this other one揭示了我想要的被称为笛卡尔积,但不是如何从字典值中得到它;同时 this one结合笛卡尔积中的列表字典,我希望输出是列表列表,如上一个链接问题,而不是字典列表。

在:

video = ["It's Friday.mp4",'Hot Koolaid.mov','The Water Buffalo Song.mp4']
CC = {'size':['6','10','14'],'font':['Courier New'],'color':['black','white'],'language':['English']}
Noise = {'CRT':['speckles','rising stripes','no signal'],'sound':['white','crackle']}

输出:

[['It's Friday.mp4','6','Courier New','black','English','speckles','white'], 
 ['Hot Koolaid.mov','6','Courier New','black','English','speckles','white']
 ...
 ['The Water Buffalo Song.mp4','14','Courier New','white','English','no signal','crackle']]

我很确定我想使用 itertools,而且我想做的是列表的笛卡尔积。我认为目前最难的事情是从字典中提取这些列表,并将它们的元素组合成列表。

________编辑:____________

在检查我随后接受的答案的过程中,我发现所有参数都在列表中很重要(为了我的目的),即使只考虑一个值也是如此;没有方括号的字符串将一次迭代一个字符。

丑陋的嵌套 for 循环看起来像:

for vid in video:
    for siz in CC['size']:
        for fon in CC['font']:
            for col in CC['color']:
                for lan in CC['language']:
                    for crt in Noise['CRT']:
                        for sou in Noise['sound']:
                            some_function(vid,siz,fon,col,lan,crt,sou)

最佳答案

字典足够小,将七个参数硬编码到 itertools.product 是最简单的:一个“独立”列表,来自 CC 的四个列表,以及来自 Noise 的两个列表。

from itertools import product
result = list(product(
                  video,
                  CC['size'],
                  CC['font'],
                  CC['color'],
                  CC['language'],
                  Noise['CRT'],
                  Noise['sound']
         ))

您可以使用 operator.itemgetter 稍微简化一下,这样可以消除对两个词典的重复提及。

from operator import itemgetter
result = list(product(
                  video,
                  *itemgetter('size', 'font', 'color', 'language')(CC),
                  *itemgetter('CRT', 'sound')(Noise)
         ))

如果您确定字典值的生成顺序,您可以进一步缩短它:

result = list(product(video, *CC.values(), *Noise.values()))

关于Python,压扁一个丑陋的嵌套for循环,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57467457/

相关文章:

python - 将字符串转换为嵌套列表

Python 列表迭代器行为和 next(iterator)

java - 将列表转换为 dict(字典)。过程桥通讯

list - 将三个列表组合成 Haskell 中的三元组列表

python - 如何检查字符串是否为空或长度为零

python - 计算其中一个引用的三个项目的最快方法(使用 python)?

python - 如何在 Python 单元测试中模拟外部服务器?

Python Numpy 获取绿色蒙版,即使其他 channel 中的像素值不全为 0

python - 有没有办法在Linux中通过Python启用tor作为系统代理设置

python - 如何在正在运行的网页上访问 SWF 中的变量值?