Установить спектрограмму фона на белый

Question

Я хотел бы установить изображение спектрограммы белым, а для остальной части мне бы хотелось, чтобы это была красная шкала.

Я пробовал plt.imshow(), используя cmap в качестве'Reds'.Но фон остается красным.Я надеюсь, что все может быть белым.

a = gen_spectrograms(my_path)
plt.imshow(a, cmap = 'Reds', vmin = 0,  alpha = 1)
plt.show()

gen_spectrograms - это функция, определяемая пользователем.

Найдены некоторые предложения, которые я должен использовать clip.Но не уверен, как это использовать.Или какими-то другими способами?

Answer 1

при условии, что ваш фон имеет значение 0, вы можете использовать set_under:

import matplotlib.pyplot as plt

cm = plt.cm.Reds
cm.set_under('white')
a=gen_spectrograms(my_path)
plt.imshow(a, cmap=cm, vmin=0.001)
plt.show()