量化投资python均线策略

均线策略是量化投资中常用的一种策略，其基本思想是通过计算一段时间内的股价平均值，来判断股票价格的走势。具体来说，当股票价格上穿均线时，说明股票价格可能会上涨，此时可以买入；当股票价格下穿均线时，说明股票价格可能会下跌，此时可以卖出或者持币观望。

以下是一个简单的均线策略的Python实现：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

# 读取股票数据
df = pd.read_csv('stock.csv')

# 计算5日均线和20日均线
df['MA5'] = df['Close'].rolling(window=5).mean()
df['MA20'] = df['Close'].rolling(window=20).mean()

# 画出股票价格和均线的走势图
plt.plot(df['Close'], label='Close')
plt.plot(df['MA5'], label='MA5')
plt.plot(df['MA20'], label='MA20')
plt.legend()
plt.show()

# 根据均线策略进行买卖
df['Signal'] = np.where(df['MA5'] > df['MA20'], 1, 0)
df['Position'] = df['Signal'].diff()

# 画出买卖信号和持仓情况的走势图
plt.plot(df['Signal'], label='Signal')
plt.plot(df['Position'], label='Position')
plt.legend()
plt.show()

在上面的代码中，我们首先读取了股票数据，然后计算了5日均线和20日均线，并画出了股票价格和均线的走势图。接着，我们根据均线策略进行买卖，当5日均线上穿20日均线时，我们认为股票价格可能会上涨，此时我们买入；当5日均线下穿20日均线时，我们认为股票价格可能会下跌，此时我们卖出或者持币观望。最后，我们画出了买卖信号和持仓情况的走势图。