pip install pandas matplotlib seaborn statsmodels yfinance

import yfinance as yf

# 下載數(shù)據(jù)（參數(shù)：股票代碼，時間范圍）
data = yf.download('600519.SS', start='2020-11-17', end='2025-11-17')
# 提取收盤價作為分析對象
close_prices = data['Close']
print(close_prices.head())

# 檢查缺失值數(shù)量
print(f"缺失值數(shù)量：{close_prices.isnull().sum()}")

# 線性插值填充缺失值
close_prices = close_prices.interpolate(method='linear')

# 確保索引為datetime類型
close_prices.index = pd.to_datetime(close_prices.index)

# 重采樣為周數(shù)據(jù)（可選）
weekly_data = close_prices.resample('W').last()

import matplotlib.pyplot as plt
plt.figure(figsize=(14, 6))
plt.plot(close_prices.index, close_prices.values, 
         label='貴州茅臺收盤價', color='#1f77b4')
plt.title('2020-2025年貴州茅臺股價走勢', fontsize=16)
plt.xlabel('日期', fontsize=12)
plt.ylabel('價格（元）', fontsize=12)
plt.grid(True, linestyle='--', alpha=0.7)
plt.legend()
plt.tight_layout()
plt.show()

# 計算20日和60日移動平均
ma_20 = close_prices.rolling(window=20).mean()
ma_60 = close_prices.rolling(window=60).mean()

plt.figure(figsize=(14, 6))
plt.plot(close_prices.index, close_prices, label='收盤價', alpha=0.5)
plt.plot(close_prices.index, ma_20, label='20日均線', linewidth=2)
plt.plot(close_prices.index, ma_60, label='60日均線', linewidth=2)
plt.title('股價與移動平均線對比', fontsize=16)
plt.legend()
plt.show()

from statsmodels.tsa.seasonal import seasonal_decompose

# 按年周期分解（假設(shè)數(shù)據(jù)有年度季節(jié)性）
result = seasonal_decompose(close_prices, model='additive', period=252)  # 252個交易日≈1年

result.plot()
plt.suptitle('股價季節(jié)性分解', y=1.02)
plt.tight_layout()
plt.show()

from statsmodels.graphics.tsaplots import plot_acf

plt.figure(figsize=(12, 6))
plot_acf(close_prices.dropna(), lags=60, alpha=0.05)  # 顯示60階自相關(guān)
plt.title('股價自相關(guān)圖', fontsize=16)
plt.xlabel('滯后階數(shù)（交易日）', fontsize=12)
plt.ylabel('自相關(guān)系數(shù)', fontsize=12)
plt.show()

pip install plotly

import plotly.graph_objects as go

fig = go.Figure()
fig.add_trace(go.Scatter(
    x=close_prices.index,
    y=close_prices,
    name='收盤價',
    line=dict(color='#1f77b4', width=2)
))

# 添加移動平均線
fig.add_trace(go.Scatter(
    x=ma_20.index,
    y=ma_20,
    name='20日均線',
    line=dict(color='#ff7f0e', width=1.5, dash='dash')
))

fig.update_layout(
    title='貴州茅臺股價交互式圖表',
    xaxis_title='日期',
    yaxis_title='價格（元）',
    hovermode='x unified',
    template='plotly_white'
)

fig.show()

# 計算滾動均值和標(biāo)準差
rolling_mean = close_prices.rolling(window=20).mean()
rolling_std = close_prices.rolling(window=20).std()

# 定義異常閾值
upper_bound = rolling_mean + 3 * rolling_std
lower_bound = rolling_mean - 3 * rolling_std

# 檢測異常值
anomalies = close_prices[(close_prices > upper_bound) | (close_prices < lower_bound)]

# 可視化
plt.figure(figsize=(14, 6))
plt.plot(close_prices.index, close_prices, label='收盤價', alpha=0.7)
plt.plot(upper_bound.index, upper_bound, 'r--', label='上界')
plt.plot(lower_bound.index, lower_bound, 'r--', label='下界')
plt.scatter(anomalies.index, anomalies, color='red', label='異常值', zorder=5)
plt.title('股價異常值檢測（3σ原則）', fontsize=16)
plt.legend()
plt.show()

# 生成完整日期范圍
full_dates = pd.date_range(start='2020-11-17', end='2025-11-17', freq='B')  # B表示工作日
close_prices = close_prices.reindex(full_dates)
close_prices = close_prices.interpolate(method='linear')  # 線性插值填充

plt.savefig('stock_analysis.png', dpi=300, bbox_inches='tight')

fig.write_html('interactive_chart.html')  # 保存為HTML
fig.write_image("interactive_chart.png", scale=2)  # 需要安裝kaleido庫

# 獲取多只股票數(shù)據(jù)
stocks = yf.download(['600519.SS', '000858.SZ', '601318.SH'], 
                    start='2020-11-17', end='2025-11-17')['Close']

# 繪制子圖
fig, axes = plt.subplots(3, 1, figsize=(14, 12))
for i, code in enumerate(stocks.columns):
    axes[i].plot(stocks.index, stocks[code], label=code)
    axes[i].set_title(f'[code]股價走勢')
    axes[i].legend()
plt.tight_layout()
plt.show()