import pandas as pd
import numpy as np
from datetime import datetime
# 假设这是从数据库中获取的历史股票数据
history_data = {
'date': ['2021-01-01', '2021-01-02', '2021-01-03'],
'open': [100, 101, 102],
'high': [102, 103, 105],
'low': [98, 99, 100],
'close': [100, 101, 102]
}
# 将获取的数据转换为DataFrame
df = pd.DataFrame(history_data)
df['date'] = pd.to_datetime(df['date'])
df.set_index('date', inplace=True)
# 计算移动平均线,例如计算5日和10日的移动平均线
df['5d_ma'] = df['close'].rolling(window=5).mean()
df['10d_ma'] = df['close'].rolling(window=10).mean()
# 计算交易信号,当10日MA上升且5日MA下降时买入,10日MA下降且5日MA上升时卖出
df['buy_signal'] = np.where(df['10d_ma'] > df['5d_ma'], 1, 0)
df['sell_signal'] = np.where(df['10d_ma'] < df['5d_ma'], 1, 0)
# 输出计算后的DataFrame
print(df)
这段代码首先导入了必要的库,并假设有一个股票的历史数据字典。然后将这些数据转换为DataFrame,并设置日期为索引。接着,它计算了5日和10日的移动平均线,并根据这些平均线计算了买入和卖出的交易信号。最后,它打印出了包含这些计算结果的DataFrame。这个例子展示了如何使用Python进行技术分析,并且是量化交易的基本步骤。