Как построить и сохранить изображение, используя Python, как в Matlab?

Question

У меня есть код в Matlab, который читает данные из текстового файла и после некоторых манипуляций с этими данными генерирует массив 20x20, который используется для построения изображения следующим образом:

Мне нужнотранскрибировать этот код в Python по причинам оптимизации и необходимости проекта.Пока фаза генерации матрицы не является правильной, но я не могу генерировать одни и те же изображения, потому что они имеют разные масштабы и цвета следующим образом:

Ниже приведен фрагмент кода в Matlab, который отображаетimage:

figure
image(H1,'CDataMapping','scaled')
colormap;
saveas(gcf,'d1.jpg');

А вот фрагмент кода в Python:

import matplotlib.pylab as plt
fig, ax = plt.subplots()
ax.imshow(H1)
plt.gcf().savefig('d1.jpg')
plt.show()
plt.close()

Мне просто нужно построить изображение с тем же разрешением (875 x 656) и изменить цвет набольше похоже.

Щас благодарю тех, кто помогает!: D

Answer 1

Разрешение контролируется двумя параметрами figsize, который принимает кортеж, представляющий размер фигуры в дюймах, а dpi контролирует число точек на дюйм фигуры. Эти двое войдут в ваш subplots аргумент:

ax, fig = plt.subplots(figsize=(x, y), dpi=z)

, где x и y - желаемая ширина и высота фигуры, а z - желаемое dpi.

Что касается цветов, вам нужно выбрать другую карту цветов . Сначала выберите нужную цветовую карту (доступные цветные карты matplotlib можно найти здесь ). Затем передайте его в аргумент cmap imshow, например:

ax.imshow(H1, cmap=c)

где c - строка, содержащая имя карты цветов.