利用python 获取股票的相关性

import tushare as ts 
ts.set_token('f17f8827bc3c6274b6810a2b2818b900447a0b6b8d359b774b1ba913') 

pro = ts.pro_api() 

from pandas import DataFrame #先建立一个字典，用来存储股票对应的价格 
all_data = {} #遍历list里面的股票，可以写入多个股票 
for ticker in ['000001.SZ','000002.SZ','000008.SZ','000004.SZ']:    #获取各股票某时段的价格    
all_data[ticker] = pro.daily(ts_code=ticker, start_date='20210101', end_date='20220120')  # 获取指定日期股票代码 #用for循环遍历股票价格并转换为dataframe的形式 
price = DataFrame({tic: data['close']            
for tic, data in all_data.items()}) #计算股票价格每日变化 
returns = price.pct_change() #计算相关性 
corr=returns.corr() 
print(corr)