Python繪制正余弦函數圖像的方法


  今天打算通過繪制正弦和余弦函數,從默認的設置開始,一步一步地調整改進,讓它變得好看,變成我們初高中學習過的圖象那樣。通過這個過程來學習如何進行對圖表的一些元素的進行調整。

  簡單繪圖

  matplotlib有一套允許定制各種屬性的默認設置。你可以幾乎控制matplotlib中的每一個默認屬性:圖像大小,每英寸點數,線寬,色彩和樣式,子圖(axes),坐標軸和網格屬性,文字和字體屬性,等等。

  安裝

  pip install matplotlib

  雖然matplotlib的默認設置在大多數情況下相當好,你卻可能想要在一些特別的情形下更改一些屬性。

  pip install matplotlib

  雖然matplotlib的默認設置在大多數情況下相當好,你卻可能想要在一些特別的情形下更改一些屬性。

  from pylab import *

  x = np.linspace(-np.pi, np.pi, 256,endpoint=True)

  C,S = np.cos(x), np.sin(x)

  plot(x,C)

  plot(x,S)

  show()

  show image

  02. 設置基本元素

  這邊的基本元素主要有幾下幾點:

  線的顏色,粗細,和線型 刻度和標簽 還有圖例

  代碼比較簡單,基本上在我的第一講內容里都講過了。

  import numpy as np

  from matplotlib import pyplot as plt

  plt.figure(figsize=(10,6), dpi=80)

  x = np.linspace(-np.pi, np.pi, 256,endpoint=True)

  C,S = np.cos(x), np.sin(x)

  # 設置線的顏色,粗細,和線型

  plt.plot(x, C, color="blue", linewidth=2.5, linestyle="-", label=r'$sin(x)$')

  plt.plot(x, S, color="red", linewidth=2.5, linestyle="-", label=r'$cos(x)$')

  # 如果覺得線條離邊界太近了,可以加大距離

  plt.xlim(x.min()*1.2, x.max()*1.2)

  plt.ylim(C.min()*1.2, C.max()*1.2)

  # 當前的刻度並不清晰,需要重新設定,並加上更直觀的標簽

  plt.xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi],

  [r'$-\pi$', r'$-\pi/2$', r'$0$', r'$+\pi/2$', r'$+\pi$'])

  plt.yticks([-1,0,1],

  [r'$-1$', r'$0$', r'$1$'])

  # 添加圖例

  plt.legend()

  plt.show()

  show image

  03. 移動軸線

  還記得我們在初高中學習的三角函數圖象,可不是這樣,它應該是有四個象限的。而這里卻是一個四四方方的圖表。

  所以接下來,我們要做的就是移動軸線,讓它變成我們熟悉的樣子。

  我們只需要兩軸線(x和y軸),所以我們需要將頂部和右邊的軸線給隱藏起來(顏色設置為None即可)。

  # plt.gca(),全稱是get current axis

  ax = plt.gca()

  ax.spines['right'].set_color('none')

  ax.spines['top'].set_color('none')

  # 由於我們移動的是左邊和底部的軸,所以不用設置這兩個也可以

  ax.xaxis.set_ticks_position('bottom')

  ax.yaxis.set_ticks_position('left')

  # 指定data類型,就是移動到指定數值

  ax.spines['bottom'].set_position(('data',0))

  ax.spines['left'].set_position(('data',0))

  關於 set_position() 這個函數中的data是啥意思?我查了下官網。解釋如下

  

在這里插入圖片描述

 

  然后最后發現,上面的寫法可以用一定更簡潔的方式設置,是等價的。

  ax.spines['bottom'].set_position('zero')

  ax.spines['left'].set_position('zero')

  show image

  04. 添加注釋

  現在的圖形部分已經成型,接下讓我們現在使用annotate命令注解一些我們感興趣的點。

  我們選擇 2π/3 作為我們想要注解的正弦和余弦值。我們將在曲線上做一個標記和一個垂直的虛線。然后,使用annotate命令來顯示一個箭頭和一些文本。

  t = 2*np.pi/3鄭州專業婦科醫院 http://www.hnzzkd.com/

  # 利用plt.plot繪制向下的一條垂直的線,利用plt.scatter繪制一個點。

  plt.plot([t,t],[0,np.cos(t)], color ='blue', linewidth=2.5, linestyle="--")

  plt.scatter([t,],[np.cos(t),], 50, color ='blue')

  plt.annotate(r'$sin(\frac{2\pi}{3})=\frac{\sqrt{3}}{2}$',

  xy=(t, np.sin(t)), xycoords='data',

  xytext=(+10, +30), textcoords='offset points', fontsize=16,

  arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))

  # 利用plt.plot繪制向上的一條垂直的線,利用plt.scatter繪制一個點。

  plt.plot([t,t],[0,np.sin(t)], color ='red', linewidth=2.5, linestyle="--")

  plt.scatter([t,],[np.sin(t),], 50, color ='red')

  plt.annotate(r'$cos(\frac{2\pi}{3})=-\frac{1}{2}$',

  xy=(t, np.cos(t)), xycoords='data',

  xytext=(-90, -50), textcoords='offset points', fontsize=16,

  arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))

  在這里,你可能會對 plt.annotate 這個函數的用法,有所陌生。這里也解釋一下。

  第一個參數,就是注釋內容; 第二個參數, xy ,就是對哪一點進行注釋; 第三個參數, xycoords ,指定類型,data 是說基於數值來定位; 第四個參數, xytext ,是注釋的位置,結合第五個參數,就是根據偏移量來決定注釋位置; 第五個參數, textcoords ,值為offset points,就是說是相對位置; 第六個參數, fontsize ,注釋大小; 第七個參數, arrowprops ,對箭頭的類型的一些設置。

  show image

  05. 完整代碼

  以上都是對片段代碼進行解釋,這里放出完整的代碼

  import numpy as np

  from matplotlib import pyplot as plt

  plt.figure(figsize=(10,6), dpi=80)

  x = np.linspace(-np.pi, np.pi, 256,endpoint=True)

  C,S = np.cos(x), np.sin(x)

  # 設置線的顏色,粗細,和線型

  plt.plot(x, C, color="blue", linewidth=2.5, linestyle="-", label=r'$sin(x)$')

  plt.plot(x, S, color="red", linewidth=2.5, linestyle="-", label=r'$cos(x)$')

  # 如果覺得線條離邊界太近了,可以加大距離

  plt.xlim(x.min()*1.2, x.max()*1.2)

  plt.ylim(C.min()*1.2, C.max()*1.2)

  # 當前的刻度並不清晰,需要重新設定,並加上更直觀的標簽

  plt.xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi],

  [r'$-\pi$', r'$-\pi/2$', r'$0$', r'$+\pi/2$', r'$+\pi$'])

  plt.yticks([-1,1],

  [r'$-1$', r'$1$'])

  # 添加圖例

  plt.legend(loc='upper left')

  # plt.gca(),全稱是get current axis

  ax = plt.gca()

  ax.spines['right'].set_color('none')

  ax.spines['top'].set_color('none')

  # 由於我們移動的是左邊和底部的軸,所以不用設置這兩個也可以

  ax.xaxis.set_ticks_position('bottom')

  ax.yaxis.set_ticks_position('left')

  # 指定data類型,就是移動到指定數值

  # ax.spines['bottom'].set_position('zero')

  ax.spines['bottom'].set_position(('data',0))

  ax.spines['left'].set_position(('data',0))

  t = 2*np.pi/3

  # 利用plt.plot繪制向下的一條垂直的線,利用plt.scatter繪制一個點。

  plt.plot([t,t],[0,np.cos(t)], color ='blue', linewidth=2.5, linestyle="--")

  plt.scatter([t,],[np.cos(t),], 50, color ='blue')

  plt.annotate(r'$sin(\frac{2\pi}{3})=\frac{\sqrt{3}}{2}$',

  xy=(t, np.sin(t)), xycoords='data',

  xytext=(+10, +30), textcoords='offset points', fontsize=16,

  arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))

  # 利用plt.plot繪制向上的一條垂直的線,利用plt.scatter繪制一個點。

  plt.plot([t,t],[0,np.sin(t)], color ='red', linewidth=2.5, linestyle="--")

  plt.scatter([t,],[np.sin(t),], 50, color ='red')

  plt.annotate(r'$cos(\frac{2\pi}{3})=-\frac{1}{2}$',

  xy=(t, np.cos(t)), xycoords='data',

  xytext=(-90, -50), textcoords='offset points', fontsize=16,

  arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))

  plt.show()

  繪制拋物線:

  X1=np.linspace(-4,4,100,endpoint=True)

  plt.plot(X1,(X1**2)/9)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM