股票python量化交易014-计算收益率

• 单只股票单次持仓收益率

计算方式：收益率 = （市价-成本)/成本*100%

收益率也叫做盈亏比，是最直接的策略收益评估指标，后面做回测也要用到。

• Matplotlib可视化库

Matplotlib 是一个Python的2D绘图库。通过 Matplotlib，开发者可以仅需要几行代码，便可以生成绘图，直方图，功率谱，条形图，错误图，散点图等。可以让数据可视化，更直观的真实给用户，使数据更加客观、更具有说服力。

简单的使用方法：

import matplotlib.pyplot as plt
df['xxx'].plot()
plt.show()

• 代码实现收益率

def calculate_prof_pct(data):
    '''
    计算单次收益率
    收益率 = （市价-成本)/成本
    :param data: 不带收益率原数据
    :return: 带收益率新数据
    '''
    data = data[data['signal'] != 0]
    data['profit_pct'] = (data['close'] - data['close'].shift(1)) / data['close'].shift(1)
    data = data[data['signal'] == -1]