pandas绘制折线图|极客笔记

pandas绘制折线图

折线图是一种常见的数据可视化方式，通过连接数据点并在两个轴上绘制线条来展示数据的变化趋势。pandas是一个功能强大的数据分析库，提供了简单易用的绘图功能，可以方便地绘制各种类型的图表，包括折线图。在本文中，我们将详细介绍如何使用pandas绘制折线图。

准备数据

在绘制折线图之前，首先需要准备数据。我们可以使用pandas的Series或DataFrame来存储数据，并将其作为绘图的输入。下面是一个示例数据集，包含了每月销售额的数据：

import pandas as pd

data = {
    'Month': ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun'],
    'Sales': [10000, 12000, 15000, 11000, 13000, 14000]
}

df = pd.DataFrame(data)
print(df)

输出为：

  Month  Sales
0   Jan  10000
1   Feb  12000
2   Mar  15000
3   Apr  11000
4   May  13000
5   Jun  14000

绘制折线图

使用pandas绘制折线图非常简单，只需要调用DataFrame的plot()方法，并指定要绘制的数据列和图表类型即可。下面是绘制月销售额折线图的示例代码：

import matplotlib.pyplot as plt

df.plot(x='Month', y='Sales', marker='o', color='b')
plt.xlabel('Month')
plt.ylabel('Sales')
plt.title('Monthly Sales')
plt.grid()
plt.show()

在这段代码中，我们首先调用DataFrame的plot()方法，指定x轴为月份列（’Month’），y轴为销售额列（’Sales’），并指定折线图的样式为圆点（marker=’o’）、颜色为蓝色（color=’b’）。然后使用plt.xlabel()和plt.ylabel()设置x轴和y轴的标签，使用plt.title()设置图表标题，最后调用plt.grid()显示网格线，并使用plt.show()展示图表。

自定义折线图样式

除了基本的折线图样式外，我们还可以通过修改参数来自定义折线图的样式，如添加图例、调整线条宽度、修改线条颜色等。下面是一个自定义折线图样式的示例代码：

df.plot(x='Month', y='Sales', marker='o', color='b', linewidth=2, label='Sales Trend')
plt.xlabel('Month')
plt.ylabel('Sales')
plt.title('Monthly Sales Trend')
plt.legend()
plt.grid()
plt.show()

在这段代码中，我们通过增加linewidth=2设置线条宽度为2，通过label='Sales Trend'添加图例，并通过plt.legend()显示图例。

多组数据折线图

除了单组数据的折线图外，我们还可以绘制多组数据的折线图，以比较它们的变化趋势。下面是一个包含两组数据的示例数据集：

data = {
    'Month': ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun'],
    'Sales1': [10000, 12000, 15000, 11000, 13000, 14000],
    'Sales2': [9000, 11000, 14000, 10000, 12000, 13000]
}

df = pd.DataFrame(data)
print(df)

输出为：

  Month  Sales1  Sales2
0   Jan   10000    9000
1   Feb   12000   11000
2   Mar   15000   14000
3   Apr   11000   10000
4   May   13000   12000
5   Jun   14000   13000

我们可以使用多次调用plot()方法来绘制多组数据的折线图，如下所示：

df.plot(x='Month', y=['Sales1', 'Sales2'], marker='o')
plt.xlabel('Month')
plt.ylabel('Sales')
plt.title('Monthly Sales Comparison')
plt.legend(['Sales1', 'Sales2'])
plt.grid()
plt.show()

在这段代码中，我们使用y=['Sales1', 'Sales2']指定同时绘制两组数据的折线图，并通过plt.legend(['Sales1', 'Sales2'])设置图例显示各组数据的含义。