def explore(wine_set):
low = wine_set[wine_set['quality'] <= 5]
medium = wine_set[(wine_set['quality'] == 6) | (wine_set['quality'] == 7)]
high = wine_set[wine_set['quality'] > 7]
print('association between wine`s density and residual sugar for wines \nof `low` quality')
print(scipy.stats.pearsonr(low['density'], low["residual_sugar"]))
print('\nof `medium` quality')
print(scipy.stats.pearsonr(medium['density'], medium["residual_sugar"]))
print('\nof `high` quality')
print(scipy.stats.pearsonr(high['density'], high["residual_sugar"]))
scat0 = seaborn.regplot(x="density", y="residual_sugar", fit_reg=True, data=low)
plt.xlabel("Density of wine")
plt.ylabel("Residual sugar in wine, gram")
plt.title("Association between wine's density and residual sugar for wines of `low` quality")
plt.show()
scat0 = seaborn.regplot(x="density", y="residual_sugar", fit_reg=True, data=medium)
plt.xlabel("Density of wine")
plt.ylabel("Residual sugar in wine, gram")
plt.title("Association between wine's density and residual sugar for wines of `medium` quality")
plt.show()
scat0 = seaborn.regplot(x="density", y="residual_sugar", fit_reg=True, data=high)
plt.xlabel("Density of wine")
plt.ylabel("Residual sugar in wine, gram")
plt.title("Association between wine's density and residual sugar for wines of `high` quality")
plt.show()
data_analysis.py 文件源码
python
阅读 34
收藏 0
点赞 0
评论 0
评论列表
文章目录