def vis_result_2D(df,label): fig = plt.figure(figsize=(16,6)) plt.rcParams['font.sans-serif'] = ['SimHei'] plt.rcParams['axes.unicode_minus'] = False ax = fig.add_subplot(121) for i in df['label'].unique(): temp_df = df[df['label']==i] ax.scatter(temp_df['WindSpeed'],temp_df['Power'],s=0.3,color=mcolors.TABLEAU_COLORS[colors[i]],label=i) plt.legend() plt.title("各类异常的分布") plt.grid(True) ax = fig.add_subplot(122) df = df[df['label']==0] ax.scatter(df['WindSpeed'],df['Power'],s=0.3) plt.title("清洗后的结果") plt.grid(True) plt.show()详细注释
时间: 2024-03-29 22:38:10 浏览: 129
matplotilb中fig,ax = plt.subplots()的使用
这是一个用于可视化数据分布的函数,输入参数为数据集 df 和标签 label。该函数会先创建一个 16x6 的画布,然后创建两个子图,分别用于显示各类异常点的分布情况和清洗后的结果。在第一个子图中,函数会根据不同标签的数据点在 WindSpeed-Power 平面上进行散点图的绘制,其中每个标签的数据点使用不同的颜色进行区分。最后,将各个标签的说明加入图例,并在图像上方添加标题。在第二个子图中,函数将仅显示标签为0的数据点,用于展示数据清洗后的结果。最后,将两个子图展示出来。
阅读全文