Как нанести несколько второстепенных сюжетов на фоновое изображение с помощью matplotlib?

1

Я хочу построить некоторые участки разброса по карте страны (изображение). Идея состоит в том, чтобы отобразить визуализацию данных области, на которой строится график.

Итак, вот как я рисую изображение карты США, где круги, которые я нарисовал в верхнем левом углу и в середине, я хотел бы отображать мои диаграммы рассеяния:

import numpy as np
import matplotlib.pyplot as plt

fig=plt.figure(figsize=(10,15))

im = plt.imread("usa-states-map.jpg")
implot = plt.imshow(im, extent=[0, 200, 0, 150])

# A circle in the upper left region
theta=np.linspace(0,2*np.pi,50)
faux_radius = 15
z=np.cos(theta)*faux_radius + 45
t=np.sin(theta)*faux_radius + 130
plt.plot(z,t)

# A circle in the middle region
theta=np.linspace(0,3*np.pi,50)
faux_radius = 15
z=np.cos(theta)*faux_radius + 100
t=np.sin(theta)*faux_radius + 80
plt.plot(z,t)

Это отображает изображение следующим образом:

Изображение 174551

Я перехожу к сюжету разброса так:

import numpy as np
import matplotlib.pyplot as plt

fig=plt.figure(figsize=(10,15))

im = plt.imread("usa-states-map.jpg")
implot = plt.imshow(im, extent=[0, 200, 0, 150])

# A circle in the upper left region
theta=np.linspace(0,2*np.pi,50)
faux_radius = 15
z=np.cos(theta)*faux_radius + 45
t=np.sin(theta)*faux_radius + 130
plt.plot(z,t)

# A circle in the middle region
theta=np.linspace(0,3*np.pi,50)
faux_radius = 15
z=np.cos(theta)*faux_radius + 100
t=np.sin(theta)*faux_radius + 80
plt.plot(z,t)

# Scatter plot 1
ax1 = plt.subplot(2,2,1)
ax1.scatter(x_1_a, y_1_a, marker="s")
ax1.scatter(x_1_b, y_1_b, marker="o")

# Scatter plot 2
ax1 = plt.subplot(2,2,2)
ax1.scatter(x_2_a, y_2_a, marker="s")
ax1.scatter(x_2_a, y_2_b, marker="o")

Но вывод, который он производит, не отображает фоновое изображение и только отображает графики рассеяния:

[Изображение 174551]

Я даже попытался использовать zorder который должен был сказать matplotlib, какой сюжет должен наступить сверху, а внизу, но безрезультатно - он производит тот же результат, что и выше:

implot = plt.imshow(im, extent=[0, 200, 0, 150], zorder=1)
...
...
...
# Scatter plot 1
ax1 = plt.subplot(2,2,1)
ax1.scatter(x_1_a, y_1_a, marker="s", zorder=2)
ax1.scatter(x_1_b, y_1_b, marker="o", zorder=2)

# Scatter plot 2
ax1 = plt.subplot(2,2,2)
ax1.scatter(x_2_a, y_2_a, marker="s", zorder=3)
ax1.scatter(x_2_a, y_2_b, marker="o", zorder=3)

Как исправить это, чтобы получить желаемый результат? Мне даже не нужны два круга, которые будут присутствовать на карте на самом деле - это было просто, чтобы проиллюстрировать, где я хотел бы построить 2 разброса. Благодарю.

Теги:
matplotlib
plot

1 ответ

2
Лучший ответ

Я смог решить проблему, используя предложение plt.axes в комментариях:

from mpl_toolkits.axes_grid.inset_locator import inset_axes
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure(figsize=(10, 15),facecolor='white')
ax = fig.add_axes([0, 0, 1, 1])
ax.axis('off')
im = plt.imread("usa-states-map.jpg")
implot = plt.imshow(im)
plt.xticks([])
plt.yticks([])

# this is an inset axes over the main axes for the top left region
a = plt.axes([.2, .6, .2, .1], facecolor='w')
plt.scatter(x_1_a, y_1_a, marker="s")
plt.scatter(x_1_b, y_1_b, marker="o")
plt.legend(['%.2f%%' %(100*len(x_1_a)/(len(x_1_a)+len(y_1_a))), '%.2f%%' %(100*len(y_1_a)/(len(x_1_a)+len(y_1_a)))], loc='upper right');

# this is an inset axes over the main axes for the middle region
a = plt.axes([.45, .45, .2, .1], facecolor='w')
plt.scatter(x_2_a, y_2_a, marker="s")
plt.scatter(x_2_b, y_2_b, marker="o")
plt.legend(['%.2f%%' %(100*len(x_2_b)/(len(x_2_b)+len(y_2_b))), '%.2f%%' %(100*len(y_2_b)/(len(x_2_b)+len(y_2_b)))], loc='upper right');

plt.show()

Изображение 174551

Ещё вопросы

Сообщество Overcoder
Наверх
Меню