时间戳

首页 > 解决方案 > Python 股票相关性热图

python - Python 股票相关性热图

问题描述

试图制作投资组合中股票的相关热图，但结果却一团糟。帮助表示赞赏。

获取此https://i.stack.imgur.com/7jzuE.jpg


pt_comp = web.DataReader(['BP.L', 'III.L', 'GSK.L', 'OCDO.L', 'RBS.L', 'SVT.L'], 'yahoo',start=start,end=end) ['Adj Close']

#Percentage returns of each stock 

pt_rets = pt_comp.pct_change()

#correlation between stocks 

corr = pt_rets.corr()

#corrlation heat map


plt.imshow(corr, cmap='hot', interpolation='none')
plt.colorbar()
plt.xticks(range(len(corr)), corr.columns)
plt.yticks(range(len(corr)), corr.columns);

'''

标签： pythonheatmapcorrelationstock

解决方案

绘制相关热图的一种方法是使用 seaborn 库。看看，有没有帮助。 https://seaborn.pydata.org/generated/seaborn.heatmap.html

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

pt_comp = web.DataReader(['BP.L', 'III.L', 'GSK.L', 'OCDO.L', 'RBS.L', 'SVT.L'], 'yahoo',start=start,end=end) ['Adj Close']

#Percentage returns of each stock 

pt_rets = pt_comp.pct_change()

#correlation between stocks 

corr = pt_rets.corr()

ax = sns.heatmap(corr)
plt.colorbar()
plt.xticks(range(len(corr)), corr.columns)
plt.yticks(range(len(corr)), corr.columns)

推荐阅读