Python Matplotlib Basemap накладывает небольшое изображение на график карты

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

самолет

Насколько я знаю и читал, pyplot.imshow() - лучший способ добиться этого. Однако я зацикливаюсь на шаге 1, заставляя изображение ровно отображаться. Используя обычный график вместо базовой карты, достаточно легко получить изображение с помощью imshow, но при использовании базовой карты я не могу заставить его отображаться вообще. См. Пример кода.

Если я могу получить изображение для отображения на карте, я предполагаю, что я могу методом проб и ошибок установить его положение и некоторые правильные размеры для него, используя атрибут extent для imshow(), с координатами графика, преобразованными из координат карты x,y = m(lons,lats) .

Вот пример кода (чтобы попробовать его, вы можете загрузить изображение самолета выше).

from matplotlib import pyplot as plt
from mpl_toolkits.basemap import Basemap
import Image
from numpy import arange

lats = arange(26,29,0.5)
lons = arange(-90,-87,0.5)

m = Basemap(projection='cyl',llcrnrlon=min(lons)-2,llcrnrlat=min(lats)-2,
            urcrnrlon=max(lons)+2,urcrnrlat=max(lats)+2,resolution='i')

x,y = m(lons,lats)
u,v, = arange(0,51,10),arange(0,51,10)
barbs = m.barbs(x,y,u,v)
m.drawcoastlines(); m.drawcountries(); m.drawstates()

img = Image.open('c130j_75px.png')
im = plt.imshow(img, extent=(x[-1],x[-1]+50000,y[-1],y[-1]+50000))
plt.show()

Вот получившееся изображение, на котором нет следов самолета. Я пробовал использовать extent несколько разных размеров, думая, что, возможно, просто сделал его слишком маленьким, но безуспешно. Я тоже пробовал установить zorder=10, но тоже безуспешно. Любая помощь будет оценена.

результат

Обновление: теперь я могу заставить изображение, по крайней мере, появиться, используя m.imshow вместо plt.imshow, поскольку первый проходит в экземпляре осей карты, но аргумент extent, похоже, не влияет на размеры изображения, поскольку он всегда заполняется весь сюжет, независимо от того, насколько маленькими я делаю extent размеров, даже если я установлю их на ноль. Как правильно масштабировать изображение самолета и расположить его рядом с последней точкой данных?

im = m.imshow(img, extent=(x[-1],x[-1]+5,y[-1],y[-1]+2))

результат2


person Levi Cowan    schedule 14.07.2012    source источник


Ответы (2)


Собственно, для этого вы хотите использовать несколько недокументированную возможность matplotlib: модуль matplotlib.offsetbox. Вот пример: http://matplotlib.sourceforge.net/trunk-docs/examples/pylab_examples/demo_annotation_box.html

В вашем случае вы бы сделали что-то вроде этого:

import matplotlib.pyplot as plt
import numpy as np
import Image

from mpl_toolkits.basemap import Basemap
from matplotlib.offsetbox import OffsetImage, AnnotationBbox

# Set up the basemap and plot the markers.
lats = np.arange(26, 29, 0.5)
lons = np.arange(-90, -87, 0.5)

m = Basemap(projection='cyl',
            llcrnrlon=min(lons) - 2, llcrnrlat=min(lats) - 2,
            urcrnrlon=max(lons) + 2, urcrnrlat=max(lats) + 2,
            resolution='i')

x,y = m(lons,lats)
u,v, = np.arange(0,51,10), np.arange(0,51,10)
barbs = m.barbs(x,y,u,v)

m.drawcoastlines()
m.drawcountries()
m.drawstates()

# Add the plane marker at the last point.
plane = np.array(Image.open('plane.jpg'))
im = OffsetImage(plane, zoom=1)
ab = AnnotationBbox(im, (x[-1],y[-1]), xycoords='data', frameon=False)

# Get the axes object from the basemap and add the AnnotationBbox artist
m._check_ax().add_artist(ab)

plt.show()

введите описание изображения здесь

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

person Joe Kington    schedule 16.07.2012
comment
Я пытаюсь использовать этот метод для анимации нескольких изображений в кадре с помощью пакета matplotlib.animation и получаю сообщение об ошибке AttributeError: 'list' object has no attribute 'axes'. Вы порекомендуете способ использования этого подхода для анимации? - person ryanjdillon; 27.03.2014
comment
Пропустили крайний срок редактирования, но я хотел бы изменить вопрос на: Вы рекомендуете способ использования этого подхода для анимации нескольких изображений в кадре? Спасибо, Джо. - person ryanjdillon; 27.03.2014
comment
Для тех, кто хочет построить несколько изображений и контролировать их zorder, используйте метод matplotlib.offsetbox.AnnotationBbox.set_zorder. - person Ioannis Filippidis; 07.06.2015
comment
На случай, если кто-то столкнется с той же проблемой, для импорта Image мне пришлось установить пакет Pillow, а затем импортировать его как from PIL import Image. - person lanadaquenada; 14.03.2019

С базовой картой вы обычно можете просто использовать обычные команды стиля pyplot, если сначала переводите свои координаты, используя экземпляр карты. В этом случае вы можете просто преобразовать экстент в uv-координаты с помощью:

x0, y0 = m(x[-1], y[-1])
x1, y1 = m(x[-1] + 0.5, y[-1] + 0.5)

И тогда впоследствии вы сможете:

im = plt.imshow(img, extent=(x0, x1, y0, y1))

Мое полное решение этой проблемы выглядит так:

import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
import numpy as np


lats = np.arange(26, 29, 0.5)
lons = np.arange(-90, -87, 0.5)

m = Basemap(projection='cyl', llcrnrlon=min(lons)-2, llcrnrlat=min(lats)-2,
            urcrnrlon=max(lons)+2, urcrnrlat=max(lats)+2, resolution='h')

x, y = m(lons,lats)
u, v = np.arange(0, 51, 10), np.arange(0, 51, 10)
barbs = m.barbs(x, y, u, v)

m.drawcoastlines()
m.fillcontinents()

x_size, y_size = 0.8, 0.4
x0, y0 = m(x[-1] - x_size/2., y[-1] - y_size/2.)
x1, y1 = m(x[-1] + x_size/2., y[-1] + y_size/2.)
im = plt.imshow(plt.imread('mslr86.png'), extent=(x0, x1, y0, y1))

plt.show()

В результате создается изображение, которое выглядит как введите описание изображения здесь

Обновление: если вы хотите, чтобы изображение оставалось фиксированного размера, независимо от увеличения, см. Ответ Джо.

person pelson    schedule 15.07.2012