import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.axisartist.axislines import Subplot
from scipy import interpolate


def sommth_plot(x_arr, y_arr):
 fig = plt.figure() # 創(chuàng)建一個figure
 ax = Subplot(fig, 111) # 利用Subplot將figure加入ax
 fig.add_axes(ax)
 ax.axis['bottom'].set_axisline_style("->", size=1.5) # x軸加上箭頭
 ax.axis['left'].set_axisline_style("->", size=1.5) # y軸加上上箭頭
 ax.axis['top'].set_visible(False) # 去除上方坐標軸
 ax.axis['right'].set_visible(False) # 去除右邊坐標軸
 xmin = min(x_arr) 
 xmax = max(x_arr)
 xnew = np.arange(xmin, xmax, 0.0005) # 在最大最小值間以間隔為0.0005插入點
 func = interpolate.interp1d(x_arr, y_arr) 
 ynew = func(xnew) # 得到插入x對應的y值
 plt.plot(xnew, ynew, '-') # 繪制圖像
 plt.show() # show圖像


if __name__ == '__main__':
 x = eval(input('輸入x：'))
 y = eval(input('輸入y：'))
 smooth_plot(x, y)

# 設置圖像標題
plt.title('title')

# 設置x范圍，y同理
plt.xlim(1, 4)

# 給x，y軸添加說明
plt.xlabel('x')
plt.ylabel('y')

# 設置線條的顏色，寬度，線條樣式，標志以及曲線的標簽
plt.plot(x, y, color='blue', linewidth=1.0, linestyle='--', marker='o', label='')
# 如果傳遞了label參量，則使用下面函數(shù)使標簽顯示，loc選擇位置，frameon=True標簽會在一個框內
plt.legend(loc='upper left', frameon=True)