如果条件为未知数量范围

时间:2018-01-22 15:13:20

标签: python arrays python-3.x numpy if-statement

我的代码为每个数字返回一个图像.eg if x==0然后它会创建一个图像并将其保存在特定的文件夹中,但问题是数字从0开始,让我们说无穷大如何避免{ {1}}在这样的问题上。 请问问题是否不清楚。在这里,我只发布了两个multiple if conditions,但它正在增加。

if condition

1 个答案:

答案 0 :(得分:0)

不要试图将所有内容命名为“第0”,“第一”,“第二”等,为什么不将文件名命名为_1.png,_2.png,...,_31.png?它允许您使代码更简单。

temp=np.zeros(shape=(100, 100),dtype=np.uint8)  
for count, (x, y) in enumerate(labels):
    component = uf.find(labels[(x, y)])
    labels[(x, y)] = component
        temp[y][x]=int(255)
        if count>32:
            temp_img = Image.fromarray(temp_img)
            temp_img.save(os.path.join(dirs, file_+'_' + str(labels[(x,y)]) +'.png'), 'png')

如果这不起作用,你可以使用一个数组或字典,其中包含所有可能性'第一','第一','第二',......,'三十二',然后使用类似的东西命名文件这个:

temp.save(os.path.join(dirs, file_ + '_'+ name_arr[labels[(x,y)]] + '.png'), 'png')