субплоты на базовую карту

Question

У меня есть карта Великобритании и 121 локация, каждая с 3 значениями. Я хотел бы построить небольшую гистограмму из трех значений в каждом из 121 местоположения.

В настоящее время значения отображаются в виде атрибута markersize, который выглядит следующим образом: Trypophobic_Plot

сделано с использованием следующего кода: (упрощенно)

fig, ax = plt.subplots()

ax = Basemap(llcrnrlat= 46.351611,
             llcrnrlon= -11.543011,
             urcrnrlat= 60.417133,
             urcrnrlon= 6.2743413,
             resolution='i', projection='tmerc', lon_0=-2, lat_0=49)

#Draw Coastlines, colours etc.
(removed because not important to this example)

################################################
# Plot Network edge map
(not important to this example)
################################################
# Plot SYNOP Station Points
lats = [59.5329945921, 58.9499976059, 57.3669959942...]
lons = [-1.63299557989, -2.90000295719, -7.40000487601...]

for method in range (0,3):
    for i in range(0, len(lons)):
        x,y = ax(lons[i], lats[i])
        ax.plot(x, y, 'o', color=(1,1,1,0), \
                markersize= ..., \
                markeredgecolor=..., \
                markeredgewidth=7)

################################################
# Other plot features
plt.text(10000, 10000,'Max: = ' ... + \
         '\nMin: = ' ..., \
         fontsize=32, zorder=75.)

plt.title('site_scores' , fontsize=30)
plt.show()

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

То, что я пробовал:

Я начал изучать преобразование между широтой / долготой и действительными осями графика, что немного сбивало с толку. Есть display, axes, data и Figure. Я не смог применить операнды преобразования здесь: https://matplotlib.org/users/transforms_tutorial.html к ax координатам. Вы можете видеть, как я делал круги в коде выше, но я не знаю, на какую систему координат их переключают.

Тогда я подумал, что попробую просто добавить ось, которую я обычно делаю, и посмотреть, где она появилась. Вот так:

ax3 = fig.add_axes([0.5,0.5, 0.2, 0.2])

но это приводит к ошибке относительно размера участка:

ValueError: Image size of 5690009x6001228 pixels is too large. It must be less than 2^16 in each direction.

Вот где я сейчас нахожусь. Я хочу небольшую линейную диаграмму размером с эти круги.

Answer 1

Код ниже использует inset_axes для построения каждой гистограммы в каждом месте на карте.Если вы установите размер figure больше, гистограммы должны быть менее загроможденными.

from mpl_toolkits.basemap import Basemap
from mpl_toolkits.axes_grid1.inset_locator import inset_axes
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches

# prep values for map extents and more
llcrnrlat = 46.351611
llcrnrlon = -11.543011
urcrnrlat = 60.417133
urcrnrlon = 6.2743413
mid_lon = (urcrnrlon+llcrnrlon)/2.0
hr_lon = (urcrnrlon-llcrnrlon)/2.0
mid_lat = (urcrnrlat+llcrnrlat)/2.0
hr_lat = (urcrnrlat-llcrnrlat)/2.0

# function to create inset axes and plot bar chart on it
# this is good for 3 items bar chart
def build_bar(mapx, mapy, ax, width, xvals=['a','b','c'], yvals=[1,4,2], fcolors=['r','y','b']):
    ax_h = inset_axes(ax, width=width, \
                    height=width, \
                    loc=3, \
                    bbox_to_anchor=(mapx, mapy), \
                    bbox_transform=ax.transData, \
                    borderpad=0, \
                    axes_kwargs={'alpha': 0.35, 'visible': True})
    for x,y,c in zip(xvals, yvals, fcolors):
        ax_h.bar(x, y, label=str(x), fc=c)
    #ax.xticks(range(len(xvals)), xvals, fontsize=10, rotation=30)
    ax_h.axis('off')
    return ax_h

fig, ax = plt.subplots(figsize=(10, 9))  # bigger is better

bm = Basemap(llcrnrlat= llcrnrlat,
             llcrnrlon= llcrnrlon,
             urcrnrlat= urcrnrlat,
             urcrnrlon= urcrnrlon,
             ax = ax,
             resolution='i', projection='tmerc', lon_0=-2, lat_0=49)

bm.fillcontinents(color='lightyellow', zorder=0)
bm.drawcoastlines(color='gray', linewidth=0.3, zorder=2)

plt.title('site_scores', fontsize=20)

# ======================
# make-up some locations
# ----------------------
n = 50   # you may use 121 here
lon1s = mid_lon + hr_lon*(np.random.random_sample(n)-0.5)
lat1s = mid_lat + hr_lat*(np.random.random_sample(n)-0.5)

# make-up list of 3-values data for the locations above
# -----------------------------------------------------
bar_data = np.random.randint(1,5,[n,3])  # list of 3 items lists

# create a barchart at each location in (lon1s,lat1s)
# ---------------------------------------------------
bar_width = 0.1  # inch
colors = ['r','y','b']
for ix, lon1, lat1 in zip(list(range(n)), lon1s, lat1s):
    x1, y1 = bm(lon1, lat1)   # get data coordinates for plotting
    bax = build_bar(x1, y1, ax, 0.2, xvals=['a','b','c'], \
              yvals=bar_data[ix], \
              fcolors=colors)

# create legend (of the 3 classes)
patch0 = mpatches.Patch(color=colors[0], label='The 0-data')
patch1 = mpatches.Patch(color=colors[1], label='The 1-data')
patch2 = mpatches.Patch(color=colors[2], label='The 2-data')
ax.legend(handles=[patch0,patch1,patch2], loc=1)

plt.show()

График будет похож на это: