我一直在尝试使用Matplotlib和Python 3在半对数坐标轴上拟合一条直线,但一直没能成功。我看过很多示例有关对数-对数比例图的内容,但我尝试过的解决方案都没有奏效(使用numpy
)。 绘制的直线总会出现扭曲。
以下是我目前所拥有的内容:
import os
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
base_path = os.path.dirname(os.path.realpath(__file__))
fig = plt.figure()
ax = fig.add_subplot(111)
# Plot data.
location = os.path.join(base_path, "data.csv")
data = np.genfromtxt(location, delimiter=',', names=['year', 'bw'])
ax.plot(data['year'], data['bw'])
# Fit test.
x = data['year']
y = data['bw']
y_ln = np.log10(y)
n = data.shape[0]
A = np.array(([[x[j], 1] for j in range(n)]))
B = np.array(y_ln[0:n])
B = np.array(y[0:n])
X = np.linalg.lstsq(A, B)[0]
a = X[0]
b = X[1]
fit = a * x + b
p = np.polyfit(x, np.log(y), 1)
ax.semilogy(x, p[0] * x + p[1], 'g--')
ax.set_yscale('log')
相关的
data.csv
文件如下所示:2016, 68.41987090116676
2017, 88.9788618486191
2018, 90.94850458504749
2019, 113.20946182004333
2020, 115.71547492850719
我得到的图形如下所示,拟合直线是弯曲的。 非常感谢反馈和建议。