使用PIL保存时修改路径名

Question

我正在从图像中提取RGB频道并将其保存为灰度PNG文件，但是我很难保存它们。这是我的代码：

listing = os.listdir(path1)    
for f in listing:
    im = Image.open(path1 + f)
    red, green, blue = im.split()
    red = red.convert('LA')
    green = green.convert('LA')
    blue = blue.convert('LA')
    red.save(path2 + f + 'r', 'png')
    green.save(path2 + f + 'g', 'png')
    blue.save(path2 + f + 'b','png')

其中path1和path2分别是图像文件夹并保存目的地。我想做的是将img.png的颜色通道的B＆amp; W版本保存到 imgr.png，imgg.png，imgb.png，但是我使用的代码是img.pngr，img.pngg，img.pngb。任何帮助将不胜感激。

Answer 1

您可以按以下方式执行此操作：

import os
listing = os.listdir(path1)    
for f in listing:
    im = Image.open(os.path.join(path1, f))
    red, green, blue = im.split()
    red = red.convert('LA')
    green = green.convert('LA')
    blue = blue.convert('LA')
    file_name, file_ext = os.path.splitext(f)
    red.save(os.path.join(path2, "{}r.png".format(file_name))
    green.save(os.path.join(path2, "{}g.png".format(file_name))
    blue.save(os.path.join(path2, "{}b.png".format(file_name))

我建议您在使用路径和文件名时使用os.path.split()和os.path.join()功能。

Answer 2

您首先需要将文件名与扩展名分开。

import os
filename = path2 + f # Consider using os.path.join(path2, f) instead
root, ext = os.path.splitext(filename)

然后，您可以再次正确组合它们：

filename = root + "r" + ext

现在filename是imgr.png而不是img.pngr。