数据格式:
    image.png

    1. # coding=utf-8
    2. import pandas as pd
    3. import seaborn as sns
    4. import matplotlib.pyplot as plt
    5. plt.rcParams['font.sans-serif']=['SimHei'] #显示中文标签
    6. plt.rcParams['axes.unicode_minus']=False #这两行需要手动设置
    7. # 读取数据
    8. io = 'data.xlsx'
    9. df = pd.read_excel(io, sheet_name = 0)
    10. df.head()
    11. # print(df.index)
    12. # print("df.head(10) :")
    13. # print(df.head(10))
    14. # print("df.describe():")
    15. # print(df.describe())
    16. # print("df['电压等级']")
    17. # print(df['电压等级'])
    18. so2 = 'so2'
    19. PM10 = 'PM10'
    20. xheight = '悬挂高度'
    21. dgrade = '电压等级'
    22. ectrifyRatio = '带电积污系数'
    23. x_ = [so2, PM10, xheight, dgrade] # 自变量
    24. y_ = [ectrifyRatio] # 因变量
    25. x__ = (so2, PM10, xheight, dgrade)
    26. y__ = (ectrifyRatio)
    27. def show_pair_plot():
    28. sns.pairplot(df, x_vars=x_, y_vars=y_, kind="reg", height=5, aspect=0.7)
    29. plt.show()
    30. # plt.savefig('img.png')
    31. # show_pair_plot()
    32. def load_data():
    33. X = df.loc[:, x__]
    34. y = df.loc[:, ectrifyRatio]
    35. print('X','\n',X)
    36. print('y','\n',y)
    37. return X,y
    38. load_data()