matplotlib 如何扩展线性回归图

mspsb9vt 于 12个月前发布在其他

关注(0)|答案(1)|浏览(75)

我遇到了一个问题，试图拟合直线到线性部分的我的图。为了完成我的图，我必须延长红线，如果它是一条直线，这样它的交叉点至少x轴可以观察到。
我的代码是：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

#data = pd.read_csv("LPPII_cw_2_1.csv")

#f = data["f [kHz]"] 
f = (1, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 200, 500)

#h21e = data["h21e [A/A]"]
h21e = (218., 215., 210., 200., 189., 175., 165., 150., 140., 129., 120., 69., 30.)

linearf = f[-3:]
linearh = h21e[-3:]

logA = np.log(linearf)
logB = np.log(linearh)

m, c = np.polyfit(logA, logB, 1, w=np.sqrt(linearh))
y_fit = np.exp(m*logA + c)

fig, ax = plt.subplots()

ax.set_xscale('log')
ax.set_yscale('log')

ax.set_xlabel('f [kHz]')
ax.set_ylabel('h$_{21e}$ [A/A]')

ax.scatter(f, h21e, marker='.', color='k')
ax.plot(linearf, y_fit, color='r', linestyle='-')

plt.show()

我的图是这样的：

matplotlib

来源：https://stackoverflow.com/questions/66755782/how-do-i-extend-a-linear-regression-plot

1条答案

按热度按时间

pgccezyw1#

您可以添加x轴的最大值并将其附加到linearf的末尾。然后计算曲线并绘制它。旧的y限制需要保存并重置，以防止matplotlib自动扩展这些限制。请注意，只有在绘制散点图后才能提取x限制。

import matplotlib.pyplot as plt
import numpy as np

f = (1, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 200, 500)
h21e = (218., 215., 210., 200., 189., 175., 165., 150., 140., 129., 120., 69., 30.)

linearf = f[-3:]
linearh = h21e[-3:]

logA = np.log(linearf)
logB = np.log(linearh)

m, c = np.polyfit(logA, logB, 1, w=np.sqrt(linearh))

fig, ax = plt.subplots()

ax.set_xscale('log')
ax.set_yscale('log')

ax.set_xlabel('f [kHz]')
ax.set_ylabel('h$_{21e}$ [A/A]')

ax.scatter(f, h21e, marker='.', color='k')

linearf_ext = list(linearf) + [ax.get_xlim()[1]]
logA = np.log(linearf_ext)
y_fit = np.exp(m * logA + c)
ymin, ymax = ax.get_ylim()
ax.plot(linearf_ext, y_fit, color='r', linestyle='-')
ax.set_ylim(ymin, ymax)
plt.tight_layout()
plt.show()

赞(0）回复(0）举报 12个月前

我来回答

matplotlib 如何扩展线性回归图

1条答案

相关问题

热门标签

最新问答