如何在python中保存.png格式的等高线图

Question

我想在python中保存轮廓。我做得不太好。我使用了plt.savefig((，但图像为空。为什么？

!pip install tftb
import numpy as np
import matplotlib.pyplot as plt
import scipy.io as sio
from tftb.generators import atoms
import tftb
import cv2
mat = sio.loadmat('/content/drive/MyDrive/z5_25.mat')
signal = mat['z']
z = signal.T
images_dir = '/content/drive/MyDrive/image'
for i in range(122):
wvd = tftb.processing.WignerVilleDistribution(z[i])
wvd.run()
fig = plt.figure()
plt.rcParams['figure.figsize']=(1.5,1.5)
wvd.plot(kind = 'contour')
plt.savefig(f"{images_dir}/fig5_25_{i}.png")
plt.show()

Answer 1

我知道已经晚了，但你可以在行中用wvd替换plt

wvd.plot(kind = 'contour') plt.savefig(f"{images_dir}/fig5_25_{i}.png")

像这样：

wvd.plot(kind = 'contour') wvd.savefig(f"{images_dir}/fig5_25_{i}.png")

它应该工作