今天用Python简单分析一下亚马逊上市至今的股价,没有太多实质性的东西,但学学技术还可以。主要包括下面几个方面:
画股价走势图
计算年度收益率
用股价拟合多项式曲线并预测股价
画K线图
首先,导入模块
import pandas as pd
import matplotlib.pylab as plt
import numpy as np
# 绘图显示中文
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
读入文件,预览数据
data = pd.read_csv('./data/Amazon.csv')
data
数据包含亚马逊1997年5月到今年7月底的股价,共5842条。
来看看上市至今的股价走势
# 整体走势
data.plot(x='Date', y='Close', rot=30, figsize=(15, 8), title='amazon股价')
这个涨势太厉害了,上市至今暴涨了1600倍。
再来看看每年的收益率,先增加年份列,然后按照该列分组,用当年的第一天和最后一天的收盘价计算涨跌幅。
# 计算每年收益率
data['year'] = data['Date'].apply(lambda x: x[:4])
def get_rate_of_change(group):
price1 = group.sort_values(by='Date')[:1]['Close']
price2 = group.sort_values(by='Date', ascending=False)[:1]['Close']
return (price2.values[0] / price1.values[0] - 1) * 100
data.groupby('year').apply(get_rate_of_change).plot