LangChain生成Count Plot：Jupyter正常VS Code无图表输出问题求助

问题场景 #

通过LangChain的create_csv_agent实现基于CSV数据集生成计数图功能，代码在Jupyter Notebook中可正常生成并显示图表，但在VS Code中直接运行时，仅输出文本结果，无任何图表显示，且无报错信息。相关代码如下：

from langchain.agents import create_csv_agent
from langchain.llms import OpenAI

import os
import pandas
os.environ["OPENAI_API_KEY"] = "Your-API-Key"

agent = create_csv_agent(OpenAI(temperature=0.2), "supermarket_sales - Sheet1.csv", verbose=True)

agent.run("Generate a count plot of Payment column?")

解决方法 #

1. 显式配置Matplotlib后端 #

VS Code默认的Matplotlib后端可能不支持非交互式环境下的图表显示，在代码开头添加以下配置，指定GUI后端：

import matplotlib
# 根据系统环境选择合适的后端，Windows可选'TkAgg'，Linux/macOS可选'Qt5Agg'
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt

2. 强制agent生成包含显示逻辑的代码 #

修改prompt，要求agent在绘图代码中明确添加显示命令：

agent.run("Generate a count plot of Payment column, and call plt.show() to display the figure?")

这样agent生成的绘图代码会包含plt.show()，触发图表显示。

3. 切换至VS Code交互式窗口运行 #

在VS Code中打开代码后，点击右上角的"运行Python文件在交互式窗口"按钮，模拟Jupyter的运行环境，该环境原生支持图表显示。

4. 验证环境依赖一致性 #

确保VS Code使用的Python环境与Jupyter Notebook一致：

• 打开VS Code终端，运行pip list查看已安装包
• 对比Jupyter中!pip list的输出，确保matplotlib、seaborn（count plot常用依赖）等包版本一致，若缺失则执行pip install matplotlib seaborn安装。

测试建议 #

可使用任意带分类列的CSV数据集测试，例如iris数据集（包含species分类列），执行类似命令：

agent.run("Generate a count plot of species column?")

验证在VS Code中是否能正常显示图表。

内容的提问来源于stack exchange，提问作者Anand Varrier