下面是一个使用pandas库来保存数据框的前5行、获取统计信息并绘制为图片输出的示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['Tom', 'Nick', 'John', 'Smith', 'Emily'],
'Age': [28, 45, 32, 19, 36],
'City': ['New York', 'Paris', 'London', 'Tokyo', 'Berlin']}
df = pd.DataFrame(data)
# 保存数据框的前5行为CSV文件
df.head(5).to_csv('output.csv', index=False)
# 获取数据框的统计信息
statistics = df.describe()
# 绘制统计信息为条形图
statistics.plot(kind='bar')
# 保存绘图结果为图片
plt.savefig('output.png')
# 显示绘图结果
plt.show()
在上面的示例中,我们首先创建了一个示例数据框df。然后,我们使用head(5)
方法选择数据框的前5行,并使用to_csv
方法将其保存为CSV文件。接下来,我们使用describe
方法获取数据框的统计信息,并将其保存在变量statistics中。然后,我们使用plot
方法将统计信息绘制成条形图,并使用savefig
方法将绘图结果保存为图片。最后,我们使用show
方法显示绘图结果。