### 设置 Python 甘特图右侧颜色注释框的实现方案 在 Python 中，无论是使用 `matplotlib` 还是 `plotly` 库绘制甘特图，都可以通过特定的方式设置右侧的颜色注释框。以下是两种常见库的具体实现方法： #### 使用 Matplotlib 实现 当使用 `matplotlib` 绘制甘特图时，可以通过 `legend` 方法创建右侧的颜色注释框。具体来说，可以利用 `Patch` 对象来定义每种颜色对应的标签。 ```python import matplotlib.pyplot as plt import matplotlib.patches as mpatches # 定义颜色列表和对应的任务名称 colors = ['r', 'pink', 'orange', 'y', 'g', 'b'] labels = ["Task A", "Task B", "Task C", "Task D", "Task E", "Task F"] # 创建 Patch 列表用于 legend patches = [mpatches.Patch(color=colors[i], label="{:s}".format(labels[i])) for i in range(len(colors))] # 添加图例 plt.legend(handles=patches, loc='best') # 显示图形 plt.show() ``` 这种方法的核心在于使用 `mpatches.Patch` 来为每种颜色创建一个补丁对象，并将其传递给 `legend` 函数以生成注释框[^3]。 #### 使用 Plotly 实现 对于 `plotly` 而言，虽然它本身不提供直接的颜色注释功能，但可以通过 `annotations` 参数手动添加说明文字并配合颜色块展示。这种方式更加灵活，适合动态交互场景。 ```python import plotly.figure_factory as ff import plotly.graph_objects as go # 示例数据 df = [ dict(Task="Job-A", Start='2023-01-01', Finish='2023-01-05', Resource="Red"), dict(Task="Job-B", Start='2023-01-06', Finish='2023-01-10', Resource="Blue") ] # 创建甘特图 fig = ff.create_gantt(df, colors=['Red', 'Blue'], index_col='Resource', show_colorbar=True) # 自定义注释框 annotation_text = "<b>Color Legend:</b><br>Red - Job Type A<br>Blue - Job Type B" fig.add_annotation( text=annotation_text, align="left", showarrow=False, xref="paper", yref="paper", x=1.05, y=1, bordercolor="#c7c7c7", borderwidth=2, borderpad=4, bgcolor="white", opacity=0.8 ) # 更新布局 fig.update_layout(width=800, height=400) fig.show() ``` 这里的关键点在于调用了 `add_annotation` 方法，在图表外部区域添加了一个自定义的文字描述框[^2]。 --- ### 总结 无论采用哪种方式，都需要根据实际需求调整样式参数。如果追求静态图像质量，则推荐使用 `matplotlib`；而如果是面向 Web 或需要更多互动特性，则建议选用 `plotly`。