Добрый день! Изучаю matplotlib и не могу понять почему функция

Question

Добрый день! Изучаю matplotlib и не могу понять почему функция

не работает
Не получается сохранить изображение в папке(plt.savefig('savefig.jpg')) и изменить его цвет ( cmap='plasma')
Что не так ?

def pics(image1, shape):
image2 = image1
# приводим изображения к одинаковой высоте или ширине в зависимости от формы коллажа
if shape == 'first':
image2[:, :, 0]
plt.imshow(image2, cmap='plasma') # отобразить изображение
#plt.savefig('savefig.jpg')
return image2