plot 用户如何将数据输入到具有最佳拟合线的图形上？

Question

如何添加最适合用户输入数据的行？ 到目前为止，我已经制作了一个循环来收集数据及其长度，但我不确定如何让它将这些值识别为 x 和 y 值的字符串，并且它需要是一个数组来做最小二乘图/最合适。 我对编码相当陌生，所以任何帮助将不胜感激，谢谢。

import matplotlib as plt
import numpy as np

x=[]
y=[]

for i in range(1, 20):
    print('Concentration?')
    x = input()
    print('Absorbance?')
    y = input()
    plt.plot(x, y, 'o', label="Data", markersize=5)
    if i == 3:
        break
    else:
        continue



A = np.vstack([x,np.ones(len(x))]).T
print(A)
#
m, c = np.linalg.lstsq(A, y, rcond=None)[0]
plt.plot(x,m*x+c, 'r', label="Fit")
plt.show()

Answer 1

您需要将 append 输入到列表中。 然后在您的 plot 中，您需要计算我在列表理解部分中显示的 y 点。

或者，您可以只使用 seaborn regplot 它将为您处理所有这些。

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

x=[]
y=[]

for i in range(1, 20):
    print('Concentration?')
    x.append(float(input()))
    print('Absorbance?')
    y.append(float(input()))
    plt.plot(x, y, 'o', label="Data", markersize=5)
    if i == 3:
        break
    else:
        continue

A = np.vstack([x,np.ones(len(x))]).T
m, c = np.linalg.lstsq(A, y, rcond=None)[0]
plt.plot(x,[m*_+c for _ in x], 'r', label="Fit")

sns.regplot(x,y, ci=None)