如何使用 Matplotlib 中的單選按鈕更改曲線?


要透過單選按鈕更改曲線的顏色,我們可以執行以下步驟 -

  • 使用 numpy 建立 x、sin 和 cos 資料點。

  • 調整子圖之間及周圍的圖形大小和填充。

  • 使用 subplots() 方法建立一個圖形和一組子圖。

  • 使用  plot() 方法繪製包含 x 和 y 資料點的曲線。

  • 使用 axes() 方法將一個軸新增到當前圖形並使其成為當前軸。

  • 將一個單選按鈕新增到當前軸。

  • 要使用單選按鈕更改曲線,我們可以使用 change_curve() 方法,該方法可以傳遞給 on_clicked() 方法中。

  • 要顯示圖形,請使用 show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.widgets import RadioButtons

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(-2, 2, 50)
sin = np.sin(x)
cos = np.cos(x)
fig, ax = plt.subplots()
l, = ax.plot(x, sin, lw=3, color='red')
rb_axis = plt.axes([0.15, 0.75, 0.15, 0.15])
radio_button = RadioButtons(rb_axis, ('sin', 'cos'))

def change_curve(c_label):
   d = {'sin': sin, 'cos': cos}
   data = d[c_label]
   l.set_ydata(data)
   plt.draw()

radio_button.on_clicked(change_curve)

plt.show()

輸出

現在,單擊單選按鈕“cos”,將更改繪圖 −

更新於:09-Apr-2021

329 次瀏覽

啟動您的 職業

完成課程即可獲得認證

開始
廣告