信号生成函数
def generate_signals(data):
data['Signal'] = 0.0
signal = 0.0
for i in range(len(data)):
if data['ShortMA'][i] data['LongMA'][i]:
signal = 1
elif data['ShortMA'][i] < data['LongMA'][i]:
signal = 0
data['Signal'][i] = signal
return data
主函数
def main():
ticker = 'AAPL' 示例股票代码
start_date = '20200101'
end_date = '20230101'
data = download_data(ticker, start_date, end_date)
data = calculate_moving_averages(data, 50, 200) 使用50日和200日移动平均线
data = generate_signals(data)
print(data)