Matplotlib - 将文本标签向右移动“x”点

Question

我有以下代码可以生成气泡图，然后将标签作为文本添加到图中：

fig, ax = plt.subplots(figsize = (5,10))

# create data
x = [1,1,1,1,1,1,1,1,1,1]
y = ['A','B','C','D',
     'E','F','G','H','I','']
z = [10,20,80,210,390,1050,2180,4690,13040,0]

labels = [1,2,8,21,39,105,218,469,1304]

plt.xlim(0.9,1.1)

for i, txt in enumerate(labels):
    ax.annotate(txt, (x[i], y[i]), ha='center', va='center', )

plt.scatter(x, y, s=z*4000, c="#8C4799", alpha=0.3)

我有垂直和水平居中的文本标签（即 1304,469 等），但理想情况下我希望它向右移动，以便远离气泡。我试过ha=right，但它只是轻推它一点点。

有什么我可以用来将它完全远离气泡的东西吗？即代码我可以把它如下for loop：

for i, txt in enumerate(labels):
    ax.annotate(txt, (x[i], y[i]), ha='center', va='center', )

Answer 1

由于s气泡的大小为s=z*4000，气泡的半径为np.sqrt(z*4000)/2。（有关解释，请参阅散点图标记大小）。

因此，您将创建一个注释，该注释位于数据坐标中气泡的中心，并np.sqrt(z*4000)/2以点为单位进行偏移（或者可能多 2 或 3 个点以使其看起来更好）。

这将使用

annotate("text", xy=(x[i],y[i]), 
         xytext=(np.sqrt(z[i]*4000)/2+2, 0),  textcoords="offset points")

完整示例：

import matplotlib.pyplot as plt
import numpy as np


fig, ax = plt.subplots(figsize = (5,10))

# create data
x = [1,1,1,1,1,1,1,1,1,1]
y = ['A','B','C','D',
     'E','F','G','H','I','']
z = [10,20,80,210,390,1050,2180,4690,13040,0]

labels = [1,2,8,21,39,105,218,469,1304]

plt.xlim(0.9,1.1)


sc = plt.scatter(x, y, s=z*4000, c="#8C4799", alpha=0.3)

for txt, size, xi, yi in zip(labels, sc.get_sizes(), x,y):
    ax.annotate(txt, xy=(xi,yi), xytext=(np.sqrt(size)/2+2, 0),
                textcoords="offset points",
                ha='left', va='center', )

plt.show()