Подтвердить что ты не робот

Определить размер оси matplotlib в пикселях

Учитывая набор axes в matplotlib, есть ли способ определить его размер в пикселях? Мне нужно масштабировать вещи в соответствии с настройкой больших или меньших цифр.

(В частности, я хочу изменить ширину линии, чтобы она была пропорциональна размеру осей.)

Ответ 1

Это дает ширину и высоту в дюймах.

bbox = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
width, height = bbox.width, bbox.height

Это, вероятно, достаточно для вашей цели, но чтобы получить пиксели, вы можете умножить на fig.dpi:

width *= fig.dpi
height *= fig.dpi

Например,

import matplotlib.pyplot as plt

def get_ax_size(ax):
    bbox = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
    width, height = bbox.width, bbox.height
    width *= fig.dpi
    height *= fig.dpi
    return width, height

fig, ax = plt.subplots()
print(get_ax_size(ax))
#(496.0, 384.00000000000006)

ax2 = plt.axes([0.3, 0.3, 0.7, 0.7])
print(get_ax_size(ax2))
# (448.0, 336.0)

Чтобы создать изображение точно такого же размера, вам нужно удалить пробел между фигурой и осью:

import numpy as np
import matplotlib.pyplot as plt

def get_ax_size(ax):
    bbox = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
    width, height = bbox.width, bbox.height
    width *= fig.dpi
    height *= fig.dpi
    return width, height

data = np.arange(9).reshape((3, 3))
fig = plt.figure(figsize=(8,6), dpi=80)
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
ax.imshow(data, aspect='equal')
print(get_ax_size(ax))
# (640.0, 480.0)
plt.savefig('/tmp/test.png', dpi=80)

% identify /tmp/test.png
/tmp/test.png PNG 640x480 640x480+0+0 8-bit DirectClass 50.5KB 0.020u 0:00.020