matplotlib 如何从图像打印中删除图例

wd2eg0qa  于 2023-04-12  发布在  其他
关注(0)|答案(1)|浏览(163)

我试图删除声谱图的图例(我试图获得244 x244像素的图像)
我试过Remove the legend on a matplotlib figure,但它的工作方式非常奇怪-我得到的结果 * 和 * 一个异常!

(我用的是Google Colab)

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
import moviepy.editor as mpy

import youtube_dl
## downloading the video
ydl_opts = {
    'format': 'bestaudio/best',
    'postprocessors': [{
        'key': 'FFmpegExtractAudio',
        'preferredcodec': 'wav',
        'preferredquality': '192',
    }],
}
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
    ydl.download(['https://www.youtube.com/watch?v=5pIpVK3Gecg'])

## selecting the audio clip from 17oth second to 180th second and saving it in talk.wav
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
ffmpeg_extract_subclip("Tyler, The Creator - 'IGOR,' Odd Future and Scoring a" 
                       "Number 1 Album _ Apple Music-5pIpVK3Gecg.wav", 170, 
                       180, targetname="talk.wav")

talk = mpy.AudioFileClip('talk.wav')
# switching axis off
plt.axis('off')

sample_rate = talk.fps
NFFT = sample_rate /25
audio_data = talk.to_soundarray()
#trying to get a 244 x 244 pixel image 
fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(2.44, 2.44), dpi=100.)

ax.axis('off')

spectrum, freqs, time, im = ax.specgram(audio_data.mean(axis=1), NFFT=NFFT, pad_to=4096, 
                                        Fs=sample_rate, noverlap=512, mode='magnitude', )

########## the problem lies here ##########
ax.get_legend.remove()

##trying to save stuff to drive but this doesnt run because of the exception
fig.colorbar(im)
fig.savefig('specto.png')

import os
print( os.getcwd() )
print( os.listdir('specto.png') )

from google.colab import files
files.download( "specto.png" )

有办法解决吗?

oewdyzsn

oewdyzsn1#

ax.get_legend引用函数对象本身,ax.get_legend()(注意括号)调用函数并返回matplotlib.legend.Legend示例或None。语法应为

ax.get_legend().remove()

但是,如果图中未添加图例,则会引发

AttributeErrorNoneType对象没有属性remove

所以最好用tryexcept子句来保护它,除非您提前知道会有一个图例。

编辑

根据注解,您混淆了matplot.pyplot.colorbar()matplotlib.pyplot.legend(),因此,当在该行之前引发异常时,它“看起来似乎可以工作”

fig.colorbar(img)

它只是从来没有绘制颜色条,因为它没有达到那条线。如果你不想要一个颜色条,然后删除线

ax.get_legend().remove()

fig.colorbar(im)

你会得到一个没有颜色条的声谱图。

相关问题