Matplotlib: Автоматично запазване на точната цифра на imshow()

Искам да създам визуализатор за 2D филтърни масиви, които се генерират от код в python. Те ще бъдат с размери от 2x2 до 7x7.
pyplot.imshow(oneFilter) ми дава чисти резултати, които искам да съхраня за всеки от 3000-те масива, които имам. Хората предлагат pyplot.imsave('onefilter.jpg', oneFilter), но резултатите се различават значително от това, което генерира imshow. Imshow ви позволява да запазите неговия резултат, който генерира следната фигура за масив 5x5: снимка, върху която щраквате запазване създава
Искам ТАЗИ снимка (още по-добре без ос, но просяците не могат да избират, предполагам) да бъде запазена автоматично.
imsave() обаче ще запази действително изображение с ширина 5x5 пиксела, което в този момент мащабът е напълно безполезен, тъй като не можете да го поставите в хартия или дори да го погледнете, за да направите извод. Освен това цветната карта изглежда се различава между двете функции. разлика imsave(вляво) и imshow(вдясно)

И така: Как да запиша графиката, която imshow() създава на моя компютър, без да я показвам на екрана? Тъй като показаната диаграма включва бутон за запазване, трябва да има метод някъде.
Опитах се да запазя резултата от imshow() в променлива, но тя връща „AxesImage“, което изглежда има няма метод за запазване.

Като алтернатива, как би продължил да генерира този вид графики по различен начин в python. Входните данни са моите 2D-масиви, изходът трябва да е нещо приятно за окото. Нетърпелив съм да науча по-добър начин.