mirror of
https://github.com/xavierchuan/FX-ML-Trading-Engine.git
synced 2026-08-09 23:50:57 +00:00
Add files via upload
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
# 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")
|
||||
Reference in New Issue
Block a user