python中matplotlib条件背景颜色的实现
Python  /  管理员 发布于 7年前   278
如何根据图表中没有的变量更改折线图的背景颜色?例如,如果我有以下数据帧:
import numpy as npimport pandas as pddates = pd.date_range('20000101', periods=800)df = pd.DataFrame(index=dates)df['A'] = np.cumsum(np.random.randn(800)) df['B'] = np.random.randint(-1,2,size=800)
如果我做df.A的折线图,如何根据该时间点'B'列的值更改背景颜色?
例如,如果在该日期B = 1,则该日期的背景为绿色。
如果B = 0,则该日期的背景应为黄色。
如果B = -1那么背景那个日期应该是红色的。
添加我最初考虑使用axvline的解决方法,但@jakevdp回答正是看起来因为不需要for循环:首先需要添加一个'i'列作为计数器,然后整个代码看起来像:
dates = pd.date_range('20000101', periods=800)df = pd.DataFrame(index=dates)df['A'] = np.cumsum(np.random.randn(800)) df['B'] = np.random.randint(-1,2,size=800)df['i'] = range(1,801)# getting the row where those values are true wit the 'i' valuezeros = df[df['B']== 0]['i'] pos_1 = df[df['B']==1]['i']neg_1 = df[df['B']==-1]['i']ax = df.A.plot()for x in zeros: ax.axvline(df.index[x], color='y',linewidth=5,alpha=0.03)for x in pos_1: ax.axvline(df.index[x], color='g',linewidth=5,alpha=0.03)for x in neg_1: ax.axvline(df.index[x], color='r',linewidth=5,alpha=0.03)
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家的支持。
122 在
学历:一种延缓就业设计,生活需求下的权衡之选中评论 工作几年后,报名考研了,到现在还没认真学习备考,迷茫中。作为一名北漂互联网打工人..123 在
Clash for Windows作者删库跑路了,github已404中评论 按理说只要你在国内,所有的流量进出都在监控范围内,不管你怎么隐藏也没用,想搞你分..原梓番博客 在
在Laravel框架中使用模型Model分表最简单的方法中评论 好久好久都没看友情链接申请了,今天刚看,已经添加。..博主 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了,可以看看近期评论的其他文章..1111 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开,博主百忙中能否发个APP下载链接,佛跳墙或极光..
Copyright·© 2019 侯体宗版权所有·
粤ICP备20027696号