我在python中遇到这个错误：ValueError：x和y必须大小相同

Question

我正在尝试导入一个csv文件，在找到线性回归后，我想为它绘制一个图形。但是在输入输出值后，它会显示我的错误。概率是正确的我已经检查过它只有这个绘图功能不起作用。任何人都可以告诉我这是什么问题。

这是我的代码：

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

def plot_regression(x, y, b): 
    # plotting the actual points as scatter plot 
    plt.scatter(x, y, color = "g", marker = "o", s = 30) 

    # predicted response vector 
    y_pred = b[0] + b[1]*x 

    # plotting the regression line 
    plt.plot(x, y_pred, color = "r") 

    # putting labels 
    plt.xlabel('x') 
    plt.ylabel('y') 

    # function to show plot 
    plt.show() 

def estimate(x,y):
    n=np.size(x)
    mx=np.mean(x)
    my=np.mean(y)

    xy=np.sum(y*x) -n*my*mx
    xx=np.sum(x*x) -n*mx*mx

    b1=xy/xx
    b0=my-b1*mx
    return(b0,b1)


data=pd.read_csv("linearwala.csv")
x=data['temp'].values
y=data['passengers'].values

b=estimate(x,y)


year=int(input("enter the value: "))

y=b[0]+b[1]*year
print("probabs are:")
print("b[0] :",b[0],"\n b[1] :",b[1])
print("final probabs are:",y)

b0=b[0]
b1=b[1]

bb=[b0,b1]
plot_regression(x,y,bb)

这是我的输出：

Here是我的csv文件。

Answer 1

你用新值覆盖你的y，因此你无法绘制它。您需要使用不同的变量名称 - 也许是prob

#...

probs=b[0]+b[1]*year
print("probabs are:")
print("b[0] :",b[0],"\n b[1] :",b[1])
print("final probabs are:",probs)

#...