matplotlib.pyplot 画多图

import matplotlib.pyplot as plt
import cv2 as cv
import os


columns=5
rows=1
fig =plt.figure(figsize=(15, 10))
image_files=r''
all_files = os.listdir(image_files)
file_list=[]
for img in all_files:
    if img.endswith('.jpg'):
        file_list.append(img)

for idx in range(rows * columns):
    
    ax = fig.add_subplot(rows, columns, idx + 1, xticks=[], yticks=[])
    img_path = os.path.join(image_files, file_list[idx])

    image = cv.imread(img_path)
        
    plt.imshow(image)

#为多个图添加一个总标题
plt.suptitle('string',fontsize=20,x=0.5,y=0.8)
plt.savefig(r'')
plt.show()

plt.close(fig)