如何将Google Colab中的matplotlib绘图代码转换为Plotly交互式绘图
Google Colab 环境下Matplotlib转Plotly交互式绘图实现
以下是完全对齐原Matplotlib绘图逻辑的Plotly版本代码,可直接在Colab中运行生成交互式绘图:
import plotly.graph_objects as go # 对齐原逻辑读取收盘价序列 close = df['A'] fig = go.Figure() # 绘制红色收盘价折线,对应原逻辑的红色粗线 fig.add_trace(go.Scatter( x=close.index, y=close.values, mode='lines', line=dict(color='red', width=2), name='收盘价' )) # 绘制signal X紫色上三角标记,对齐原markevery参数的筛选逻辑 fig.add_trace(go.Scatter( x=close.index[df_x], y=close.values[df_x], mode='markers', marker=dict(symbol='triangle-up', color='magenta', size=10), name='signal X' )) # 绘制signal Y黑色下三角标记,对齐原markevery参数的筛选逻辑 fig.add_trace(go.Scatter( x=close.index[df_y], y=close.values[df_y], mode='markers', marker=dict(symbol='triangle-down', color='black', size=10), name='signal Y' )) # 对齐原逻辑的标题、画布尺寸、图例配置 fig.update_layout( title=f'Turtle Agent: total gains {df_A}, total investment {df_B}%', width=1500, height=500 ) # Colab环境直接运行即可展示交互式绘图 fig.show()
- 生成的交互图支持悬停看具体数值、局部缩放、图例点击显隐对应序列等功能
- 如果你的
df_x/df_y是布尔类型的掩码序列,直接将close.index[df_x]替换为close.index[df_x.astype(bool)]即可正常适配 - 所有样式参数完全对齐原Matplotlib的绘图效果,无需额外调整
内容的提问来源于stack exchange,提问作者Red
相关产品推荐
相关产品推荐

