import numpy as np
import matplotlib.pyplot as plt

# 創(chuàng)建一個(gè)簡(jiǎn)單的數(shù)組
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 使用 Matplotlib 繪制圖形
plt.plot(x, y)
plt.title('Sine Wave')
plt.xlabel('Time')
plt.ylabel('Amplitude')
plt.show()

# 創(chuàng)建兩個(gè)不同的數(shù)據(jù)系列
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

# 繪制兩個(gè)數(shù)據(jù)系列
plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend()  # 顯示圖例
plt.show()

# 創(chuàng)建隨機(jī)數(shù)據(jù)
x = np.random.randn(100)
y = np.random.randn(100)

# 繪制散點(diǎn)圖
plt.scatter(x, y)
plt.title('Scatter Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.show()

# 創(chuàng)建隨機(jī)數(shù)據(jù)
data = np.random.randn(1000)

# 繪制直方圖
plt.hist(data, bins=30, alpha=0.7)
plt.title('Histogram')
plt.xlabel('Value')
plt.ylabel('Frequency')
plt.show()

# 創(chuàng)建網(wǎng)格數(shù)據(jù)
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sqrt(X**2 + Y**2)

# 繪制等高線圖
plt.contourf(X, Y, Z, 20, cmap='viridis')
plt.colorbar()
plt.title('Contour Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.show()

# 創(chuàng)建一個(gè)隨機(jī)的二維數(shù)據(jù)矩陣
data = np.random.rand(10, 10)

# 繪制熱力圖
plt.imshow(data, cmap='hot', interpolation='nearest')
plt.colorbar()
plt.title('Heatmap')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.show()

# 創(chuàng)建分類數(shù)據(jù)
labels = ['A', 'B', 'C', 'D']
sizes = [15, 30, 45, 10]

# 繪制餅圖
plt.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140)
plt.title('Pie Chart')
plt.show()

# 創(chuàng)建隨機(jī)數(shù)據(jù)
data = np.random.randn(100)

# 繪制箱線圖
plt.boxplot(data)
plt.title('Box Plot')
plt.ylabel('Value')
plt.show()

from mpl_toolkits.mplot3d import Axes3D

# 創(chuàng)建三維數(shù)據(jù)
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 創(chuàng)建三維圖形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, Z, cmap='viridis')

ax.set_title('3D Surface Plot')
ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
ax.set_zlabel('Z-axis')
plt.show()

import matplotlib.animation as animation

# 創(chuàng)建數(shù)據(jù)
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 創(chuàng)建圖形對(duì)象
fig, ax = plt.subplots()
line, = ax.plot(x, y)

# 更新函數(shù)
def update(frame):
    line.set_ydata(np.sin(x + frame / 10.0))
    return line,

# 創(chuàng)建動(dòng)畫
ani = animation.FuncAnimation(fig, update, frames=100, interval=50)
plt.title('Dynamic Sine Wave')
plt.xlabel('Time')
plt.ylabel('Amplitude')
plt.show()

import pandas as pd
import yfinance as yf

# 下載股票數(shù)據(jù)
ticker = 'AAPL'
data = yf.download(ticker, start='2022-01-01', end='2023-01-01')

# 繪制股票價(jià)格走勢(shì)圖
plt.figure(figsize=(12, 6))
plt.plot(data['Close'], label='Close Price')
plt.title(f'{ticker} Stock Price')
plt.xlabel('Date')
plt.ylabel('Price (USD)')
plt.legend()
plt.show()

# 計(jì)算移動(dòng)平均線
data['MA50'] = data['Close'].rolling(window=50).mean()
data['MA200'] = data['Close'].rolling(window=200).mean()

# 繪制移動(dòng)平均線
plt.figure(figsize=(12, 6))
plt.plot(data['Close'], label='Close Price')
plt.plot(data['MA50'], label='50-Day MA')
plt.plot(data['MA200'], label='200-Day MA')
plt.title(f'{ticker} Stock Price with Moving Averages')
plt.xlabel('Date')
plt.ylabel('Price (USD)')
plt.legend()
plt.show()