plt.imsave() 的“格式”參數不起作用

Question

根據文檔，當format參數不在fname參數中時，它可以指定文件擴展名。

format str, optional 文件格式，例如'png', 'pdf', 'svg', ... 未設置時的行為記錄在fname 下。

但是，它對我不起作用，它輸出一個沒有文件擴展名的文件。

import numpy as np
import os
import matplotlib.pyplot as plt

pic = np.random.randint(0, 256, (224, 224, 3), dtype=np.uint8)

plt.imsave('test', pic, format='jpg')

os.listdir()

['test']

如何在不指定fname的情況下保存 JPG？

Answer 1

format參數只是改變文件格式，與文件名無關。 如果您測試 bash 中的文件類型，即使文件名不包含“.jpg”，它也應該顯示正確的格式：

$ file test
test: JPEG image data, JFIF standard 1.01, ...

如果您在fname中定義擴展名，則不需要format ：

plt.imsave('test.jpg')

如果您想在不輸入兩次的情況下明確表示，可以將擴展名抽象為變量：

ext = 'jpg'
plt.imsave(f'test.{ext}', pic, format=ext)