大家好，想要改善以下程式，讓它漂亮一點但不知道怎麼寫
str1=('D:/ANN/training/0/*png','D:/ANN/training/1/*png',
'D:/ANN/training/2/*png','D:/ANN/training/3/*png',
'D:/ANN/training/4/*png')
coll0 = io.ImageCollection(str1[0])
mat0 = io.concatenate_images(coll0)
coll1 = io.ImageCollection(str1[1])
mat1 = io.concatenate_images(coll1)
coll2 = io.ImageCollection(str1[2])
mat2 = io.concatenate_images(coll2)
coll3 = io.ImageCollection(str1[3])
mat3 = io.concatenate_images(coll3)
coll4 = io.ImageCollection(str1[4])
mat4 = io.concatenate_images(coll4)
不知道這段程式該如何改善?有查到要用dict，但還是不知道如何使用
資料裡面都是PNG檔，謝謝