python - 如何摆脱 python 中的多个嵌套循环

Question

这是我的代码；

instance_count = 10
for imgs in wire_images:
    for counter in range(0, instance_count, 1):
        for oimgs in images_path:
            applyWireAugmentation(oimgs, imgs, wire_dir, odir, 0, dst_dir, counter, "waug")


def applyWireAugmentation(img, wire_img,wdir, odir, theata,dst_path, counter, index):

    src_im = Image.open(wdir+wire_img).convert("LA")
    dst_im = Image.open(odir+img)
    w,h = dst_im.size
    angle = theata
    size = 200, h

    x = random.randint(0,w)
    y = 0
    im = src_im.convert('RGBA')
    rot = im.rotate( angle, expand=1 ).resize(size)
    dst_im.paste( rot, (x, y), rot )
    dst_im.save(dst_path+"/"+img.replace(".png","")+"-"+index+"-"+str(counter)+".png")

wire_images 包含文件夹中的所有图像文件。 images_path 将所有图像文件放在另一个文件夹中。我需要从 wire_images 中获取一张图像并将其应用于 oimgs instance_count 次(在本例中为 10 次)。是否有任何圆滑的 pythonic 方法来摆脱这些循环或使其更快？

Answer 1

你之前可以生成所有可能的组合，但我不确定它是否更好:

import itertools
comb = itertools.product(wire_images, images_path, range(0,instance_count,1))

for imgs, oimgs, counter in comb:
    applyWireAugmentation(oimgs, imgs,wire_dir,odir, 0, dst_dir, counter,  "waug")