在Google Colab运行W3School代码遇AttributeError错误求助

问题背景 #

在Google Colab中运行W3School提供的Matplotlib绘图代码（包括替换为印度GDP数据集后），触发以下错误：

AttributeError: 'OutStream' object has no attribute 'buffer'

错误根源是代码中的plt.savefig(sys.stdout.buffer)语句。

问题原因 #

W3School的这段代码是为适配其自有在线编译器环境编写的——通过将图像输出到标准输出缓冲区来实现绘图显示。但Google Colab的sys.stdout实现逻辑不同，它的OutStream对象并不具备buffer属性，因此这两行适配代码在Colab中完全冗余，甚至会引发错误。

此外，Colab默认已配置好Matplotlib的显示后端，无需手动设置matplotlib.use('Agg')；原代码中health_data.plot(...)末尾的逗号会导致返回元组，可能引发潜在问题。

解决方案 #

1. 移除代码末尾的plt.savefig(sys.stdout.buffer)和sys.stdout.flush()两行
2. 可选：移除import matplotlib和matplotlib.use('Agg')（Colab默认支持Matplotlib图像显示）
3. 移除health_data.plot(...)末尾的逗号
4. 根据实际数据集调整坐标轴范围（原范围不适合GDP数据）

修正后的完整代码示例 #

import pandas as pd
import matplotlib.pyplot as plt

# 读取Colab中的数据集（路径请根据自身文件位置调整）
health_data = pd.read_csv("/content/drive/MyDrive/India_GDP_Data.csv", header=0, sep=",")

# 绘制折线图，注意移除末尾逗号
health_data.plot(x='Year', y='GDP_In_Billion_USD', kind='line')

# 根据印度GDP数据调整坐标轴范围
plt.ylim(ymin=0, ymax=3000)
plt.xlim(xmin=1960, xmax=2020)

# 直接在Colab中显示图像
plt.show()

额外说明 #

在Colab中，执行plt.show()即可自动在单元格下方渲染图像。若需要保存图像到文件，可使用plt.savefig("gdp_plot.png")，图像会保存到Colab当前工作目录，也可指定完整路径如plt.savefig("/content/drive/MyDrive/gdp_plot.png")。

内容的提问来源于stack exchange，提问作者Siddhant Prajapati