关于legend的补充

发布于: 雪球转发:1回复:0喜欢:0
感谢柯克船长 无私分享的学习python的系列文章。
在《小白学量化投资(10):计算股票相关性(上)》,网页链接 中,将两只股票的收盘价进行归一化处理并打印出来的时候,两条纠缠在一起的线,缺少对应的Legend,不好分清哪条线对应的是哪支股票。如下图:



如果做些小改进,将两条线合并到一个新的DataFrame,并启新的列标签:
df=pd.concat([df.gd_one,df.pf_one],axis=1,keys=[stock1,stock2]) ,
之后再打印,就可显示Legend,可以更方便看清哪条线对应的是哪支股票。如下图:


这样改可以吗?请教@柯克船长 [大笑]