mirror of
https://github.com/xavierchuan/FX-ML-Trading-Engine.git
synced 2026-07-28 10:37:44 +00:00
xiaochuan
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
# scripts/plot_candles.py
|
|
|
|
import os
|
|
import pandas as pd
|
|
import plotly.graph_objects as go
|
|
|
|
BASE_DIR = os.path.dirname(os.path.dirname(__file__))
|
|
RAW_DATA_DIR = os.path.join(BASE_DIR, "data", "raw")
|
|
|
|
def plot_csv_candles(file_path, symbol, granularity):
|
|
df = pd.read_csv(file_path, parse_dates=["time"])
|
|
|
|
fig = go.Figure(data=[
|
|
go.Candlestick(
|
|
x=df["time"],
|
|
open=df["open"],
|
|
high=df["high"],
|
|
low=df["low"],
|
|
close=df["close"]
|
|
)
|
|
])
|
|
fig.update_layout(
|
|
title=f"{symbol} {granularity} Candlestick Chart",
|
|
xaxis_title="Time",
|
|
yaxis_title="Price",
|
|
xaxis_rangeslider_visible=False
|
|
)
|
|
|
|
# 创建 charts 文件夹（如果不存在）
|
|
os.makedirs("charts", exist_ok=True)
|
|
output_file = f"charts/{symbol.replace('/', '')}_{granularity}.html"
|
|
fig.write_html(output_file)
|
|
print(f"✅ Saved chart to {output_file}")
|
|
|
|
if __name__ == "__main__":
|
|
plot_csv_candles(os.path.join(RAW_DATA_DIR, "EURUSD_H1.csv"), "EUR/USD", "H1")
|