Как нарисовать наклонные треугольники в matplotlib?

Question

Я хотел бы иметь треугольники, показанные на изображении:

Вот мой код:

import matplotlib.pyplot as plt
data= [0.2855,0.3030,0.4995]
x = [1,2,3]
plt.plot(x, data)
plt.show

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

Answer 1

В зависимости от вашей идеи «автоматической моды», это может быть подходящим решением:

import matplotlib.pyplot as plt
import numpy as np

# Data
x = np.array([1, 2, 3])
y = np.array([0.2855, 0.3030, 0.4995])

# Calculate triangle coordinates values
x_mid = np.convolve(x, [0.5, 0.5], mode='valid')
x_tri = np.vstack((x_mid, x_mid + 0.3))
y_tri = np.interp(x_tri, x, y)

# Calculate slopes
slopes = np.diff(y) / np.diff(x)

# Plot
plt.plot(x, y)
plt.plot(x_tri, np.tile(y_tri[0, :], [2, 1]), 'r')      # red horizontal line
plt.plot(np.tile(x_tri[1, :], [2, 1]), y_tri, 'r')      # red vertical line
for i, slope in enumerate(slopes):                      # slope values
    plt.text(x_tri[1, i] + 0.05, np.mean(y_tri[:, i]), r'{0:.3f}'.format(slope))
plt.show()

Вывод:

Поместите все элементы треугольника в отдельную функцию, и это не сильно повлияет на ваш основной код.

Надеюсь, это поможет!