在曲线中求直线部分和斜率

Question

对于一个监测时间函数质量变化的装置，我们想要计算数据的线性部分的斜率。

下面所示的示例是通过读取设备产生的数据而产生的。

import pandas as pd
import matplotlib.pyplot as plt

#Find DataFrame
df = pd.read_table("raw_data.csv", sep =";", skiprows = 11, decimal = ",")

# Plot figures
plt.figure()
plt.plot(df["Time(s)"], df["Mass(g)"], label = "Raw Data")
plt.axvspan(2, 17, color="b", alpha=0.2)
plt.xlabel("Time (s)")
plt.ylabel("Mass (g)")
plt.legend(loc=0)
plt.axis([0, None, 0, None])
plt.show()

​

​

是否可以在这条曲线中拟合线性部分(大致是突出显示的部分)并计算其斜率？

Answer 1

使用numpy.polyfit()：

df_sampled = df[:max_value] #select the points you want to keep
m, p = numpy.polyfit(df_sampled.index, df_sampled, deg=1)

该函数返回线性回归的斜率和截取。