Python读取mat(matlab数据文件)并画图。
代码:
import matplotlib.pyplot as plt
import numpy as np
import scipy.io
plt.rcParams['axes.unicode_minus'] = False #显示负号
data_snr = scipy.io.loadmat(r'C:\Users\Administrator\Desktop\mat1\angle1.mat') # 横坐标数据,读取mat文件
data_RMSE1 = scipy.io.loadmat(r'C:\Users\Administrator\Desktop\mat1\Pmusic1.mat') # 纵坐标数据一,读取mat文件
data_RMSE2 = scipy.io.loadmat(r'C:\Users\Administrator\Desktop\mat1\Pmusic2.mat') # 纵坐标数据二,读取mat文件
data_RMSE3 = scipy.io.loadmat(r'C:\Users\Administrator\Desktop\mat1\Pmusic3.mat') # 纵坐标数据二,读取mat文件
#print(data_snr.keys()) # 查看mat文件中的所有变量
snr=data_snr['angle1']
print(snr)
R1list=data_RMSE1['Pmusic1'].tolist()
R2list=data_RMSE2['Pmusic2'].tolist()
R3list=data_RMSE3['Pmusic3'].tolist()
snrlist = snr.tolist()
snrlist = snrlist[0]
print(snrlist[-1]) #横坐标最右值
d=30 #横坐标间隔
list1 = R1list[0]
list2 = R2list[0]
list3 = R3list[0]
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
plt.xlabel("Angle of incidence/(degree)")
plt.ylabel("Spatial spectrum/(dB)")
x = snrlist
#x[0] = -90
my_x_ticks = np.arange(-90, snrlist[-1]+1, d)
plt.xticks(my_x_ticks)
plt.plot(x, list1, label='Basic_MUSIC',color="k", linestyle='--',linewidth=1) # 添加label设置图例名称
plt.plot(x, list2, label='TOP_Denoise',color="k", linestyle='-',linewidth=0.6) # 添加label设置图例名称
plt.plot(x, list3, label='TOP_Iter',color="k", linestyle='-.',linewidth=0.6) # 添加label设置图例名称
plt.legend()
plt.show()