Python 股票市场分析实战

目标：

1.股票数据获取

2.历史趋势分析及可视化

3.风险分析

实验数据：来源于Yahoo Finance / Stooq，该网站提供了很多API接口，本文用的工具是pandas-datareader

实验平台：Anaconda （pandas-datareader 需手动pip install）

导入工具包

import datetime
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import pandas_datareader as web

获取数据

start = datetime.datetime(2016, 1, 1)  # 获取数据的时间段-起始时间
end = datetime.date.today()  # 获取数据的时间段-结束时间
alibaba = web.DataReader("BABA", "stooq", start, end)  # 获取阿里巴巴2016年1月1日至今的股票数据

查看数据

print(alibaba.head())  # 查看前5条数据

print(alibaba.tail())  # 查看后5条数据
print(alibaba.index)  # 查看DataFrame索引，索引为时间序列
print(alibaba.columns)  # 查看DataFram列名，列信息为开盘价、最高价、最低价、收盘价、成交量
print(alibaba.shape)  # 查看DataFram形状
print(alibaba.describe())  # 查看DataFrame数据每组的统计情况，如最小值、最大值、均值、标准差等

趋势分析及可视化

alibaba["Close"].plot(legend=True)  # 收盘价可视化
plt.show()

alibaba["Volume"].plot(legend=True)  # 交易量可视化
plt.show()
 

# 每日内股票变化
alibaba['high-low'] = alibaba['High'] - alibaba['Low']
print(alibaba.head())

# 每日间股票变化daily return
alibaba['daily-return'] = alibaba['Close'].pct_change()
alibaba['daily-return'].plot(figsize=(40,2))
plt.show()

# 每日间股票变化分布图
alibaba['daily-return'].plot(kind='hist')
plt.show()

sns.distplot(alibaba['daily-return'].dropna())
plt.show()

风险分析

# 获取五家科技公司2015至今的股票数据
start = datetime.datetime(2015,1,1)
company = ['AAPL','GOOG','MSFT','AMZN','FB']
top_tech_df = web.DataReader(company,'stooq',start)['Close']
print(top_tech_df.head())

1 # 每日收盘价较前一个交易日的百分比变化，即涨跌幅度
2 top_tech_dr = top_tech_df.pct_change()
3 print(top_tech_dr.head())

# 五家公司的股票收盘价走势图
top_tech_df.plot()
plt.show()

# 探究两组变量的分布及关系
sns.jointplot(x='GOOG',y='MSFT',data=top_tech_dr)
plt.show()

  

 # 所有股票相关性比较

sns.pairplot(top_tech_dr.dropna())
plt.show()