pandas画图如何保存下来|极客笔记

pandas画图如何保存下来

在数据分析过程中，经常会用到pandas库来进行数据处理和分析。其中，pandas提供了一些简单易用的绘图功能，可以帮助我们直观地展示数据的分布和趋势。在进行数据分析的过程中，我们常常需要保存绘制的图表，以便后续的报告、分享或展示。本文将详细介绍如何使用pandas绘制图表，并将其保存下来。

1. pandas绘图基础

pandas提供了一个简单的绘图功能，可以通过plot()方法轻松地绘制各种类型的图表，如折线图、柱状图、散点图等。在使用前，需要先导入pandas库，并将数据转换为DataFrame格式。

import pandas as pd

# 创建一个示例数据
data = {'A': [1, 2, 3, 4, 5],
        'B': [5, 4, 3, 2, 1]}

df = pd.DataFrame(data)

# 绘制折线图
df.plot()

上面的代码中，我们首先导入pandas库，并创建了一个示例数据，然后将数据转换为DataFrame格式。接着使用DataFrame的plot()方法绘制了一个简单的折线图。

2. 保存绘图

pandas绘制的图表默认是显示在Jupyter Notebook或其他编辑器中的，如果我们想要将其保存下来，可以通过两种方式实现：使用savefig()方法或在plot()方法中设置参数。

2.1 使用`savefig()`方法保存图表

在绘制完图表后，我们可以调用savefig()方法将图表保存为图片文件。在调用savefig()方法时，可以指定保存文件的格式（如PNG、JPEG、PDF等）和文件名。

import pandas as pd

# 创建一个示例数据
data = {'A': [1, 2, 3, 4, 5],
        'B': [5, 4, 3, 2, 1]}

df = pd.DataFrame(data)

# 绘制折线图
ax = df.plot()
fig = ax.get_figure()
fig.savefig('line_plot.png')  # 保存为PNG格式的图片文件

在上面的代码中，我们利用get_figure()方法获取图表对应的Figure对象，并通过调用savefig()方法将图表保存为PNG格式的图片文件。

2.2 在`plot()`方法中设置参数保存图表

除了使用savefig()方法保存图表外，我们还可以在plot()方法中直接设置参数来保存图表。通过设置filepath_or_buffer参数，可以将绘制的图表保存为指定的路径下的文件。

import pandas as pd

# 创建一个示例数据
data = {'A': [1, 2, 3, 4, 5],
        'B': [5, 4, 3, 2, 1]}

df = pd.DataFrame(data)

# 绘制柱状图并保存为文件
df.plot(kind='bar', filepath_or_buffer='bar_plot.png')  # 保存为PNG格式的图片文件

在上面的代码中，我们将kind参数设置为'bar'，表示绘制柱状图，并将filepath_or_buffer参数设置为文件路径，使得绘制的图表可以直接保存为PNG格式的图片文件。