如何在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()

输出：

将所有三角形的内容放在单独的函数中，不会对您的主代码产生太大影响。

希望有帮助！