优化:如何改进if语句?

时间:2018-04-27 01:02:19

标签: python if-statement optimization

我有多个图像,里面有一个形状。我想将这个形状装箱以便以后裁剪这些图像。我希望每个图像都有相同的裁剪框(形状之间的坐标变化不大)。

我得到了这段代码:

from PIL import Image
import glob
import numpy as np

def Region():

    path = glob.glob('label/*.tif')
    L = [640,640,0,0]
    for img_name in path:
        x = []
        y = []
        img = Image.open(img_name)
        pic = np.array(img)

        for index, lit in enumerate(pic):
            for indox, value in enumerate(lit):
                if value == 0:
                    y.append(index)
                    x.append(indox)

        if min(y) < L[0]:
            L[0]=min(y)
        if min(x) < L[1]:
            L[1]=min(x)
        if max(y) > L[2]:
            L[2]=max(y)
        if max(x) > L[3]:
            L[3]=max(x)

   print L

它给了我这个输出[298,258,424,386],它与我想要的完全一致。 但很好我需要处理大量数据,我相信可能有更好的方法来做到这一点。

由于我在编程方面相当新,我想知道是否有更好的方法来完成这项任务。

有任何建议吗?

非常感谢,

此致

编辑:如果陈述L [3]':'

1 个答案:

答案 0 :(得分:0)

您可以在循环期间而不是在循环之后确定最小值和最大值。

    for index, lit in enumerate(pic):
        for indox, value in enumerate(lit):
            if value == 0:
                if index < L[0]:
                    L[0] = index
                if indox < L[1]:
                    L[1] = indox
                if index > L[2]:
                    L[2] = index
                if indox > L[3]:
                    L[3] = indox