优化图表颜色为红涨绿跌
This commit is contained in:
13
backtest.py
13
backtest.py
@@ -119,6 +119,17 @@ def load_strategy(strategy_file):
|
||||
return calculate_indicators, strategy_class
|
||||
|
||||
|
||||
def apply_color_scheme():
|
||||
"""
|
||||
应用颜色方案:红涨绿跌(中国股市风格)
|
||||
"""
|
||||
import backtesting._plotting as plotting
|
||||
from bokeh.colors.named import tomato, lime
|
||||
|
||||
plotting.BULL_COLOR = tomato
|
||||
plotting.BEAR_COLOR = lime
|
||||
|
||||
|
||||
def parse_arguments():
|
||||
"""
|
||||
解析命令行参数
|
||||
@@ -256,6 +267,8 @@ def main():
|
||||
# 解析参数
|
||||
args = parse_arguments()
|
||||
|
||||
apply_color_scheme()
|
||||
|
||||
# 加载数据
|
||||
print(f"加载股票数据: {args.code} ({args.start_date} ~ {args.end_date})")
|
||||
data = load_data_from_db(args.code, args.start_date, args.end_date)
|
||||
|
||||
Reference in New Issue
Block a user