折线图
plt.figure(figsize=(40, 40)) # 确定图像画布的大小
plt.subplot(211) # 将画布分为两行一列
plt.xlabel('Number of sample', fontsize=40) # x轴的label
plt.ylabel('Characteristics of the amplitude', fontsize=40) # y轴的label 备注（plot所有的原件都可以加fontsize属性）
plt.title('{} characteristics (ml_id=2 waveType=2)'.format(c_type), fontsize=50) # 图的title
plt.plot(two_type_list[:two_negative_end_index], linestyle = "-", color = 'r', # 绘制折线图，其中若x参数省略，则横坐标以y列表的索引代替
label = 'Negative | average: {} variance: {} median: {}'.format(('%.2f' % np.mean(two_type_list[ : two_negative_end_index])), # label参数表示这条线的label，可以当作图例显示出来
('%.2f' % np.var(two_type_list[ : two_negative_end_index])),
('%.2f' % np.median(two_type_list[ : two_negative_end_index]))),
linewidth=3.0) # 线宽
plt.plot(two_type_list[two_negative_end_index+1:], linestyle = "-", color = 'g', # 备注（一张图可以累积加多个plot）
label = 'Positive | average: {} variance: {} median: {}'.format(('%.2f' % np.mean(two_type_list[two_negative_end_index+1 : ])),
('%.2f' % np.var(two_type_list[two_negative_end_index+1 : ])),
('%.2f' % np.median(two_type_list[two_negative_end_index+1 : ]))),
linewidth=3.0)
# plt.ylim(0, 5) # 设置y轴的取值范围，如设置（0,5）则y轴坐标为从0开始，到5结束
# 刻度值字体大小设置
plt.tick_params(labelsize=40) # 设置坐标轴上刻度的字体大小
plt.legend(loc=0, fontsize = 40) # 显示图例，loc=0表示图例会根据图片情况自动摆放
####################################################################################################################################
plt.subplot(212)
plt.xlabel('Number of sample', fontsize=40)
plt.ylabel('Characteristics of the amplitude', fontsize=40)
plt.title('{} characteristics (ml_id=6 waveType=2)'.format(c_type), fontsize=50)
plt.plot(six_type_list[:six_negative_end_index], linestyle = "-", color = 'r',
label = 'Negative | average: {} variance: {} median: {}'.format(('%.2f' % np.mean(six_type_list[ : six_negative_end_index])),
('%.2f' % np.var(six_type_list[ : six_negative_end_index])),
('%.2f' % np.median(six_type_list[ : six_negative_end_index]))),
linewidth=3.0)
plt.plot(six_type_list[six_negative_end_index+1:], linestyle = "-", color = 'g',
label = 'Positive | average: {} variance: {} median: {}'.format(('%.2f' % np.mean(six_type_list[six_negative_end_index+1 : ])),
('%.2f' % np.var(six_type_list[six_negative_end_index+1 : ])),
('%.2f' % np.median(six_type_list[six_negative_end_index+1 : ]))),
linewidth=3.0)
# 刻度值字体大小设置
plt.tick_params(labelsize=40)
plt.legend(loc=0, fontsize = 40)
plt.savefig('C:/Users/Mloong/Desktop/f_image/{} characteristics.png'.format(c_type), dpi=300)
plt.show()
2.散点图
_type = 'median'
plt.scatter(range(0, 3790), two_avgAbs_list[0:3790], c='r') # 散点图的x参数不可省略
plt.scatter(range(3791, 4939), two_avgAbs_list[3791:4939], c='g')
plt.title('{} ml_id=2 waveType=2'.format(_type))
plt.savefig('C:/Users/Mloong/Desktop/f_image/{} scatter ml_id=2 waveType=2.png'.format(_type), dpi=300)
plt.show()
3.概率分布图
# 概率分布图
type_list = two_median_list
_type = 'median'
num_bins = 100 # 条状图的个数
plt.hist(type_list[:3790], num_bins, normed=1, facecolor='blue', alpha=0.5)
plt.hist(type_list[3791:], num_bins, normed=1, facecolor='red', alpha=0.5)
plt.xlabel('Value')
plt.ylabel('Probability')
plt.title('{} probability distribution ml_id=2 waveType=2'.format(_type))
plt.subplots_adjust(left=0.15)
plt.savefig('C:/Users/Mloong/Desktop/f_image/{} probability distribution ml_id=2 waveType=2.png'.format(_type), dpi=300)
plt.show()
4.箱形图
_type = 'pca_value'
import seaborn as sns
plt.subplot(121)
plt.title('{} (ml_id=2 waveType=2)'.format(_type))
sns.set(style='whitegrid') # 设置背景
sns.boxplot(x='label', y='{}'.format(_type), data=two_data, hue='label') # data参数是一个dataframe对象，其中x和y分别时这个dataframe中的列名
#########################################################################################
plt.subplot(122)
plt.title('{} (ml_id=6 waveType=2)'.format(_type))
sns.set(style='whitegrid') # 设置背景
sns.boxplot(x='label', y='{}'.format(_type), data=six_data, hue='label') # 绘制箱形图
plt.savefig('C:/Users/Mloong/Desktop/f_image/{} box figure.png'.format(_type), dpi=300)
plt.show()
5.热图
# 2.相关矩阵
import seaborn as sns
corrmat = two_data[['avs', 'avgAbs', 'rms', 'rms2', 'wave', 'pulse', 'PeekFlag',
'Margin', 'Skewness', 'Kurtosis', 'median', 'pca_value', 'label']].corr() # .corr()求相关矩阵，此时返回的值corrmat为相关矩阵
f, ax = plt.subplots(figsize=(12, 9))
sns.heatmap(corrmat, vmax=.8, square=True) # 将这个相关矩阵以热图的形式画出来
plt.savefig('C:/Users/Mloong/Desktop/f_image/two correlation matrix.png', dpi=300)
plt.show()
