数据格式:
# coding=utf-8import pandas as pdimport seaborn as snsimport matplotlib.pyplot as pltplt.rcParams['font.sans-serif']=['SimHei'] #显示中文标签plt.rcParams['axes.unicode_minus']=False #这两行需要手动设置# 读取数据io = 'data.xlsx'df = pd.read_excel(io, sheet_name = 0)df.head()# print(df.index)# print("df.head(10) :")# print(df.head(10))# print("df.describe():")# print(df.describe())# print("df['电压等级']")# print(df['电压等级'])so2 = 'so2'PM10 = 'PM10'xheight = '悬挂高度'dgrade = '电压等级'ectrifyRatio = '带电积污系数'x_ = [so2, PM10, xheight, dgrade] # 自变量y_ = [ectrifyRatio] # 因变量x__ = (so2, PM10, xheight, dgrade)y__ = (ectrifyRatio)def show_pair_plot():sns.pairplot(df, x_vars=x_, y_vars=y_, kind="reg", height=5, aspect=0.7)plt.show()# plt.savefig('img.png')# show_pair_plot()def load_data():X = df.loc[:, x__]y = df.loc[:, ectrifyRatio]print('X','\n',X)print('y','\n',y)return X,yload_data()
