Показывать текстовые аннотации только над строкой, а не под

Question

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

Как сделать так, чтобы каждая аннотация отображалась над линией, связанной с ее точкой серой линией? Некоторые из них уже такие, а другие нет.

x = list(df_usa['date'][61:-1].str.slice(5,)) # strings
x_range = np.arange(0, len(x))
y = list(df_usa['growth_factor'][61:-1])

plt.figure(figsize=(20,6))
plt.plot(x, y, color="red", alpha=0.5)
plt.xticks(np.arange(0, len(x), 5))

texts = [plt.text(x_range[i], y[i], round(y[i], 2)) for i in range(len(x))]

f = interpolate.interp1d(x_range, y)
f_x = np.arange(min(x_range), max(x_range), 0.005)
f_y = f(f_x)

# Generate non-overlapping data point labels
adjust_text(texts, x=f_x, y=f_y,
            only_move={'points': 'y', 'text': 'y'},
            force_points=0.15,
            autoalign='y',
            arrowprops=dict(arrowstyle="-", color="gray", lw=0.5))

plt.show()

Answer 1

Текущий подход «запрещает» точки на кривой. Расширением могло бы быть также запретить полосу точек под кривой.

Поскольку ваше сообщение, по-видимому, не содержит тестовых данных, приведена адаптация примера связанного сообщения:

import matplotlib.pyplot as plt
from adjustText import adjust_text
import numpy as np
from scipy import interpolate

together = [(0, 1.0, 0.4), (25, 1.0127692669427917, 0.41), (50, 1.016404709797609, 0.41), (75, 1.1043426359673716, 0.42), (100, 1.1610446924342996, 0.44), (125, 1.1685687930691457, 0.43), (150, 1.3486407784550272, 0.45), (250, 1.4013999168008104, 0.45)]
together.sort()

text = [x for (x,y,z) in together]
eucs = [y for (x,y,z) in together]
covers = [z for (x,y,z) in together]

p1 = plt.plot(eucs,covers,color="black", alpha=0.5)
y0, y1 = plt.ylim()
plt.ylim(y0, y1+.1*(y1-y0))
texts = []
for x, y, s in zip(eucs, covers, text):
    texts.append(plt.text(x, y, s))

f = interpolate.interp1d(eucs, covers)
x = np.arange(min(eucs), max(eucs), 0.0005)
y = f(x)

# optionally show the band of "forbidden" points
# plt.scatter(x=np.tile(x, 5), y=np.concatenate([y-eps for eps in np.linspace(0, 0.01, 5) ]), alpha=.1)

plt.xlabel("Proportional Euclidean Distance")
plt.ylabel("Percentage Timewindows Attended")
plt.title("Test plot")
adjust_text(texts, x=np.tile(x, 5), y=np.concatenate([y-eps for eps in np.linspace(0, 0.01, 5) ]), autoalign='y',
            only_move={'points':'y', 'text':'y'}, force_points=.3,
            arrowprops=dict(arrowstyle="->", color='r', lw=0.5))
plt.show()