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()

% matlab
xlabel('x軸');
ylabel('y軸');
title('標題');
lengend('x','y','z')

# python
import matplotlib.pyplot as plt
plt.rcParams["font.sans-serif"]=["Microsoft YaHei"] #設置字體,防止中文和負號亂碼

plt.xlabel('x軸') # 設置x軸
plt.ylabel('y軸') # 設置y軸
plt.title('標題')