代码之家  ›  专栏  ›  技术社区  ›  Lynn

人为增加一个值并观察对另一个值的影响-Python

  •  0
  • Lynn  · 技术社区  · 4 年前

    我有两列,我想人为地将A列的值增加到1000,看看B列的值会发生什么。

    数据

    A           B
    500         20
    200         10
    100         5
    

    渴望的

    A           B
    500        20
    200        10
    100        5
    1000       ?
    

    我想人为地将A列的值增加到1000,看看B列的值会发生什么变化。

    正在执行

    使用python,我将测试相关性。 将其视为线性回归问题。

    pyplot.scatter(x = ‘A’, y = ‘B’, s= 100)
    pyplot.show()
    

    然后我想,如果我增加A的依赖值,我可以使用线性回归来确定B的值。只是不确定如何输入假设值。

    import numpy as np
    from sklearn.linear_model import LinearRegression
    
    x = np.array([500,200,100]).reshape((-1, 1))
    y = np.array([20,10,5])
    

    欢迎提出任何建议

    1 回复  |  直到 4 年前
        1
  •  1
  •   OctopuSS7    4 年前

    在使用模型进行预测之前,您首先需要创建并拟合模型。

    import numpy as np
    from sklearn.linear_model import LinearRegression
    x = np.array([500,200,100])
    y = np.array([20,10,5])
    reg = LinearRegression().fit(x, y)
    reg.predict(np.array([1000]))
    

    图表可能会有所帮助。在这种情况下,不存在严格的线性关系,但我们正在进行最佳猜测。这有点像一台电脑在画一条最合适的线。

    enter image description here

    这里,最佳拟合线的方程是 Y = 0.03654*X + 1.923 。进行预测只意味着在这个公式中插入另一个X值,以找到相应的Y坐标。