【題目描述】設
，
，
，其中
，完成下列操作：
（1）在同一座標系下用不同的顏色和線型繪製y1、y2和y3三條曲線；
（2）在同一繪圖框內以子圖形式繪製y1、y2和y3三條曲線。
【練習要求】請給出原始碼程式和執行測試結果，原始碼程式要求新增必要的註釋。
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 10, 0.0001)
y1 = x ** 2
y2 = np.cos(x * 2)
y3 = y1 * y2
def plot_first_graph():
plt.figure()
plt.plot(x, y1, linestyle='-.', label='y1 = x^2')
plt.plot(x, y2, linestyle=':', label='y2 = cos(2x)')
plt.plot(x, y3, linestyle='--', label='y3 = y1 * y2')
plt.legend()
plt.savefig("3-1.png")
plt.show()
def plot_second_graph():
fig, subs = plt.subplots(2, 2)
subs[0, 0].plot(x, y1)
subs[0, 0].set_title('y1 = x^2')
subs[0, 1].plot(x, y2)
subs[0, 1].set_title('y2 = cos(2x)')
subs[1, 0].plot(x, y3)
subs[1, 0].set_title('y3 = y1 * y2')
plt.tight_layout()
plt.savefig("3-2.png")
plt.show()
while True:
print("請選擇以下選項:")
print("1. 繪製第一個圖")
print("2. 繪製第二個圖")
print("其他數字: 退出")
choice = input("請輸入選項: ")
if choice == '1':
plot_first_graph()
elif choice == '2':
plot_second_graph()
else:
break
