我想绘制时间序列面积图,其中正 (>= 0) 值用一种颜色填充,负 (< 0) 值用另一种颜色填充。
以这个例子为例:
import pandas as pd
import numpy as np
import plotly.express as px
df = pd.read_csv('https://raw.githubusercontent.com/plotly/datasets/master/finance-charts-apple.csv').assign(
PnL = lambda x: x['AAPL.Close'] - 100
)
px.area(
data_frame = df,
x = 'Date',
y = 'PnL',
width = 500,
height = 300
)
我希望 PnL 低于 0 的部分填充为红色。
所以这就是我的尝试:
import pandas as pd
import numpy as np
import plotly.express as px
df = pd.read_csv('https://raw.githubusercontent.com/plotly/datasets/master/finance-charts-apple.csv').assign(
PnL = lambda x: x['AAPL.Close'] - 100
)
df['sign'] = np.where(df['PnL'] >= 0, 'positive', 'negative')
px.area(
data_frame = df,
x = 'Date',
y = 'PnL',
color = 'sign',
color_discrete_map = {
'positive': 'steelblue',
'negative': 'crimson'
},
width = 500,
height = 300
)
但这给了我:
这不是我要找的。执行此操作的最佳方法是什么?
最佳答案
这是我在有耐心的时候能做的最好的事情:
import plotly.graph_objects as go
mask = df['PnL'] >= 0
df['PnL_above'] = np.where(mask, df['PnL'], 0)
df['PnL_below'] = np.where(mask, 0, df['PnL'])
fig = go.Figure()
fig.add_trace(go.Scatter(x=df['Date'], y=df['PnL_above'], fill='tozeroy'))
fig.add_trace(go.Scatter(x=df['Date'], y=df['PnL_below'], fill='tozeroy'))
结果:
显然不理想,但可以帮助您完成大部分工作。两条迹线相交处有一些轻微的伪影,显然当值为零时线条颜色仍然可见。
通过在两条轨迹中添加mode='none'
,可以去掉线条,只渲染填充区域:
关于Python Plotly Express : How to conditionally fill an area plot?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69039149/