如何生成结果作为图形标题以及与绘制图像相关的文件名？

Question

我需要帮助，将测试结果和文件名放在一起作为评估图像的标题。我可以将测试结果生成为列表（参见图片），我可以将测试结果添加到图像上。我可以将文件名添加到图像顶部。但是，我不能两者兼得。不确定我做错了什么。谢谢。path = '/Users/minny/Desktop/A/png/file2/'ref_images = glob.iglob(path + "*.png")  all_ref_images = []  for ref_path in ref_images:      ref_head, tail = os.path.splitext(ref_path)     image1 = Image.open(ref_path)     imgA= print(calculate_brightness(image1))    #all_ref_images.append([imgA, ref_head])        ref_image = plt.imshow(image1)    fig = plt.figure()            plt.title(os.path.basename(ref_head)), plt.title(calculate_brightness(image1))            #plt.axis("off")def calculate_brightness(image):greyscale_image = image.convert('L')histogram = greyscale_image.histogram()pixels = sum(histogram)brightness = scale = len(histogram)for index in range(0, scale):    ratio = histogram[index] / pixels    brightness += ratio * (-scale + index)return 1 if brightness == 255 else brightness / scale%%capture #gives a standard size image to view inline with the text def display_img(img, cmap=None):     fig = plt.figure(figsize=(3,3))     ax = fig.add_subplot(111)     ax.imshow(img, cmap)  path = '/Users/minny/Desktop/A/png/file2/' ref_images = glob.iglob(path + "*.png")   all_ref_images = []  for ref_path in ref_images:      ref_head, tail = os.path.splitext(ref_path)     image1 = Image.open(ref_path)     imgA= print(calculate_brightness(image1))    all_ref_images.append([imgB, ref_head])                   fig = plt.figure()          ref_image = plt.imshow(image1)    print(os.path.basename(ref_head))            #plt.axis("off")    ref_image = plt.imshow(image1)    image_basename = os.path.basename(ref_head)    title = '\n'.join([image_basename, str(calculate_brightness(image1))])    plt.title(title, loc='left')    dir_name = '/Users/minny/Desktop/A/png/results/'    plt.savefig('{dir_name}/{filename}'.format(dir_name=dir_name, filename=image_basename))

Answer 1

您可以将图像路径的基本名称与calculate_brightness函数的输出连接起来，并将结果设置为标题而不覆盖它们：

ref_image = plt.imshow(image1)

image_basename = os.path.basename(ref_head)

test_results = '\n'.join(map(str, calculate_brightness(image1)))

title = '\n'.join([image_basename, test_results])

plt.title(title, loc='left')

UPD：如果calculate_brightness函数的结果是浮点数，你可以通过这种方式解决你的问题：

ref_image = plt.imshow(image1)

image_basename = os.path.basename(ref_head)

title = '\n'.join([image_basename, str(calculate_brightness(image1))])

plt.title(title, loc='left')

UPD2： 要将图像保存到指定文件夹，您可以使用 plt.savefig 方法：

dir_name = '/Users/minny/Desktop/A/png/file2/some_directory' # create directory if necessary

plt.savefig('{dir_name}/{filename}'.format(dir_name=dir_name, filename=image_basename))