如何在自动放置的散点箭头上注释点

Question

如果我用 matplotlib 制作散点图：

plt.scatter(randn(100),randn(100))
# set x, y lims
plt.xlim([...])
plt.ylim([...])

我想用指向给定点 (x, y) 的箭头和标签来注释它。我知道这可以通过注释来完成，但我希望箭头及其标签以“最佳”方式放置，如果可能的话（给定当前轴比例/限制）箭头和标签不与其他点重叠。例如，如果您想标记异常点。有办法做到这一点吗？它不必是完美的，而只是箭头/标签的智能放置，仅给出要标记的点的 (x,y) 坐标。谢谢。

Answer 1

基本上，不，没有。

处理与此类似的地图标签放置的布局引擎非常复杂，超出了 matplotlib 的范围。 （边界框交叉实际上是决定标签放置位置的一种相当糟糕的方法。为只能在 1000 种情况中的一种情况下工作的东西编写大量代码有什么意义？）

除此之外，由于复杂的数量matplotlib 所做的文本渲染（例如 Latex），如果不先完全渲染文本，就不可能确定文本的范围（这相当慢）。

然而，在许多情况下，您会发现在带有注释的标签后面使用透明框是一个合适的解决方法。

例如

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(1)
x, y = np.random.random((2,500))

fig, ax = plt.subplots()
ax.plot(x, y, 'bo')

# The key option here is `bbox`. I'm just going a bit crazy with it.
ax.annotate('Something', xy=(x[0], y[0]), xytext=(-20,20), 
            textcoords='offset points', ha='center', va='bottom',
            bbox=dict(boxstyle='round,pad=0.2', fc='yellow', alpha=0.3),
            arrowprops=dict(arrowstyle='->', connectionstyle='arc3,rad=0.5', 
                            color='red'))

plt.show()

Answer 2

使用 adjustText （完全公开，我写的）。

让我们标记前 10 个点。我更改的唯一参数是降低点的排斥力，因为点的数量太多，我们希望算法花费更多时间并更仔细地放置注释。

import numpy as np
import matplotlib.pyplot as plt
from adjustText import adjust_text
np.random.seed(1)
x, y = np.random.random((2,500))

fig, ax = plt.subplots()
ax.plot(x, y, 'bo')
ts = []
for i in range(10):
    ts.append(plt.text(x[i], y[i], 'Something'+str(i)))
adjust_text(ts, x=x, y=y, force_points=0.1, arrowprops=dict(arrowstyle='->', 
color='red'))
plt.show()

这并不理想，但这里的点非常密集，有时无法将文本放置在目标附近而不与其中任何一个重叠。但这都是自动的且易于使用，并且不会让标签相互重叠。

聚苯乙烯
它使用边界框交叉点，但我想说相当成功！

Answer 3

另一个使用基于 Phlya 包的示例/latest/Examples.html#Now-a-classical-mtcars-dataset;-example-from-ggrepel-package-for-R" rel="nofollow noreferrer">调整Text_mtcars：

from adjustText import adjust_text
import matplotlib.pyplot as plt
                                                                                                                                
mtcars = pd.read_csv(
    "https://gist.githubusercontent.com/seankross/a412dfbd88b3db70b74b/raw/5f23f993cd87c283ce766e7ac6b329ee7cc2e1d1/mtcars.csv"
)
                                                                                                                                
def plot_mtcars(adjust=False, force_points=1, *args, **kwargs):
    # plt.figure(figsize=(9, 6))
    plt.scatter(mtcars["wt"], mtcars["mpg"], s=15, c="r", edgecolors=(1, 1, 1, 0))
    texts = []
    for x, y, s in zip(mtcars["wt"], mtcars["mpg"], mtcars["model"]):
        texts.append(plt.text(x, y, s, size=9))
    plt.xlabel("wt")
    plt.ylabel("mpg")
    if adjust:
        plt.title(
            "force_points: %.1f\n adjust_text required %s iterations"
            % (
                force_points,
                adjust_text(
                    texts,
                    force_points=force_points,
                    arrowprops=dict(arrowstyle="-", color="k", lw=0.5),
                    **kwargs,
                ),
            )
        )
    else:
        plt.title("Original")
    return plt
                                                                                                                                
fig = plt.figure(figsize=(12, 12))
                                                                                                                                
force_points = [0.5, 1, 2, 4]
for index, k in enumerate(force_points):
    fig.add_subplot(2, 2, index + 1)
    plot_mtcars(adjust=True, force_points=k)