从纬度/经度坐标计算像素值（使用matplotlib Basemap）

Question

我需要将地图坐标转换为像素（以便在HTML中制作可点击的地图）。

这是一个示例地图（使用matplotlib的Basemap软件包制作）。 我在上面放了一些标签，并尝试计算标签的中点像素：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

## Step 0: some points to plot
names = [u"Reykjavík", u"Höfn", u"Húsavík"]
lats = [64.133333, 64.25, 66.05]
lons = [-21.933333, -15.216667, -17.316667]

## Step 1: draw a map using matplotlib/Basemap
from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt

M = Basemap(projection='merc',resolution='c',
            llcrnrlat=63,urcrnrlat=67,
            llcrnrlon=-24,urcrnrlon=-13)

x, y = M(lons, lats) # transform coordinates according to projection
boxes = []
for xa, ya, name in zip(x, y, names):
    box = plt.text(xa, ya, name,
        bbox=dict(facecolor='white', alpha=0.5))
    boxes.append(box)

M.bluemarble() # a bit fuzzy at this resolution...
plt.savefig('test.png', bbox_inches="tight", pad_inches=0.01)

# Step 2: get the coordinates of the textboxes in pixels and calculate the
# midpoints
F = plt.gcf() # get current figure
R = F.canvas.get_renderer()
midpoints = []
for box in boxes:
    bb = box.get_window_extent(renderer=R)
    midpoints.append((int((bb.p0[0] + bb.p1[0]) / 2),
            int((bb.p0[1] + bb.p1[1]) / 2)))

这些计算出的点之间的相对关系大致正确，但与真实点不重合。以下代码片段应该在每个标签的中心放置一个红色圆点：

# Step 3: use PIL to draw dots on top of the labels
from PIL import Image, ImageDraw

im = Image.open("test.png")
draw = ImageDraw.Draw(im)
for x, y in midpoints:
    y = im.size[1] - y # PIL counts rows from top not bottom
    draw.ellipse((x-5, y-5, x+5, y+5), fill="#ff0000")
im.save("test.png", "PNG")

• 红点应该在标签的中间。

我认为错误出现在提取文本框坐标时（步骤#2）。非常感谢任何帮助。

注

• 也许解决方案类似于这个答案？

- Michael Dunn

---

你能否使用Basemap来绘制红点呢？请参考http://matplotlib.org/basemap/api/basemap_api.html#mpl_toolkits.basemap.Basemap.plot - tommy.carstensen

Answer 1

你的像素位置偏差是由两个因素引起的。

1. 用于计算文本位置的dpi与保存图形时使用的dpi不同。

2. 当你在savefig调用中使用bbox_inches选项时，它会消除很多白色空间。你在使用PIL绘制圆形时没有考虑到这一点（或者检查当别人点击时也是如此）。此外，在这个savefig调用中添加了一个填充，如果非常大（如我在下面的示例中所示），则可能需要考虑这个填充。如果仍然使用0.01，则可能不要紧。

要解决第一个问题，只需强制图形和savefig调用使用相同的DPI。

要解决第二个问题，请记录坐标轴在像素中的(0,0)位置，并相应地移动文本位置。

下面是稍微修改过的代码:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

## Step 0: some points to plot
names = [u"Reykjavík", u"Höfn", u"Húsavík"]
lats = [64.133333, 64.25, 66.05]
lons = [-21.933333, -15.216667, -17.316667]

## Step 1: draw a map using matplotlib/Basemap
from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt

# predefined dpi
FIGDPI=80

# set dpi of figure, so that all calculations use this value
plt.gcf().set_dpi(FIGDPI)

M = Basemap(projection='merc',resolution='c',
            llcrnrlat=63,urcrnrlat=67,
            llcrnrlon=-24,urcrnrlon=-13)

x, y = M(lons, lats) # transform coordinates according to projection
boxes = []
for xa, ya, name in zip(x, y, names):
    box = plt.text(xa, ya, name,
        bbox=dict(facecolor='white', alpha=0.5))
    boxes.append(box)

M.bluemarble() # a bit fuzzy at this resolution...

# predefine padding in inches
PADDING = 2
# force dpi to same value you used in your calculations
plt.savefig('test.png', bbox_inches="tight", pad_inches=PADDING,dpi=FIGDPI)

# document shift due to loss of white space and added padding
origin = plt.gca().transAxes.transform((0,0))
padding = [FIGDPI*PADDING,FIGDPI*PADDING]

第二步未变

第三步考虑了来源

# Step 3: use PIL to draw dots on top of the labels
from PIL import Image, ImageDraw

im = Image.open("test.png")
draw = ImageDraw.Draw(im)
for x, y in midpoints:
    #  deal with shift
    x = x-origin[0]+padding[0]
    y = y-origin[1]+padding[1]
    y = im.size[1] - y # PIL counts rows from top not bottom
    draw.ellipse((x-5, y-5, x+5, y+5), fill="#ff0000")
im.save("test.png", "PNG")

这将导致：

请注意，我使用了夸张的 PADDING 值来测试一切是否仍然有效，而 0.01 的值将产生您的原始图。

- Yann

---

太棒了！非常感谢您提供完整、清晰且经过测试的答案。 - Michael Dunn