only size-1 arrays can be converted to Python scalars和Polynomial must be 1d only

python中在使用plt.bar()画条形图和parameter = np.polyfit(y_test, y_pred, 1)进行趋势线函数参数时经常会遇到如上两种错误。

不仅如此,由于初学python对list和numpy还有dataframe不熟悉,在变量使用上经常出现各种各样的问题,都可以尝试使用这种方法将数据转为dataframe统一类型,然后再根据dateframe的列进行切片,这样不仅可以保证数据的一致性,在对二维数据的增删改查中dataframe比其他两种也更加方便。

y_pred = pd.DataFrame((y_pred)) # 先将数据转为dataframe类型
parameter = np.polyfit(y_test, y_pred.iloc[:, 0], 1) # 利用iloc切片,也可以直接data['列名']进行,iloc中逗号前为行切片,后面为列切片,我选择的是第一列的所有数据
f = np.poly1d(parameter)  # 拼接方程
plt.plot(y_test, f(y_test), "r", linewidth=0.3, label='Trendline')
plt.plot(y_test, y_pred, '*y') # 这样出图就不会报错

另外数据维度等错误由于找不到当时的错误提示了,也是用这个方法解决的,目前还是比较方便的一种方法。

这个问题困扰了我好久,今天终于解决了,之前我都是把数据导出为txt再loadtxt,很麻烦又存很多txt系统文件,加油,慢慢熟悉python。

你可能感兴趣的:(python,python,numpy,机器学习)