如何使用先前的预测模型来预测一系列数字之间的值

Question

问：为了更好地理解模型，我们可以可视化其预测。使用您的第一个模型来预测18至80岁之间每个整数年龄段的收入。您的预测结果应存储在称为pred_age的numpy浮点数组中。

我的模特：

  a1, b1 = np.polyfit(df["age"],df["income"],1) 

我做了什么：

我使用以下代码获取18-80的值，然后将其存储到18-80岁的x中

x = np.array(df.loc[(df['age'] >= 18) & (df['age'] <= 80), ['age']],)

然后我做了：

pred_age = a1 * x + b1

print(pred_age)

断言错误： Assertion Error Message

Problem Screenshot