如何用Matplotlib为Pandas DataFrame绘制折线图？|极客笔记

如何用Matplotlib为Pandas DataFrame绘制折线图？

Matplotlib是Python中最流行的可视化库之一，它可以帮助我们创建各种类型的图形，包括折线图。在这篇文章中，我们将介绍如何使用Matplotlib为Pandas DataFrame绘制折线图。

Pandas DataFrame

Pandas是一个广泛使用的Python数据分析库，它提供了很多功能，包括数据清洗、聚合、重塑和分析。其中最常用的数据结构是DataFrame，它类似于电子表格，可以将数据以表格的形式存储在其中。下面是一个示例DataFrame：

import pandas as pd

data = {'year': [2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019],
        'sales': [100, 200, 150, 300, 400, 350, 500, 550, 600, 700]}
df = pd.DataFrame(data)
print(df)

输出如下：

   year  sales
0  2010    100
1  2011    200
2  2012    150
3  2013    300
4  2014    400
5  2015    350
6  2016    500
7  2017    550
8  2018    600
9  2019    700

Matplotlib折线图

接下来，我们将使用Matplotlib为上述DataFrame绘制折线图。Matplotlib提供了一个pyplot模块，它类似于MATLAB的绘图工具箱，可以帮助我们快速创建各种类型的图形。

我们可以使用plot()函数来创建折线图，该函数接受一个x轴数据的数组和一个y轴数据的数组。下面是一个生成折线图的示例代码：

import matplotlib.pyplot as plt

plt.plot(df['year'], df['sales'])
plt.show()

添加标题和标签

Matplotlib还可以帮助我们自定义折线图，例如添加标题、轴标签和图例。下面是示例代码：

import matplotlib.pyplot as plt

plt.plot(df['year'], df['sales'])
plt.title('Sales by Year')
plt.xlabel('Year')
plt.ylabel('Sales (Millions)')
plt.legend(['Sales'])
plt.show()

这将输出一个带有标签、标题和图例的折线图：

改变线条样式和颜色

我们还可以添加线条样式和颜色，以使折线图更具吸引力。Matplotlib提供了许多不同的线条样式和颜色，可以帮助我们自定义折线图。下面是示例代码：

import matplotlib.pyplot as plt

plt.plot(df['year'], df['sales'], color='red', linestyle='--', linewidth=2)
plt.title('Sales by Year')
plt.xlabel('Year')
plt.ylabel('Sales (Millions)')
plt.legend(['Sales'])
plt.show()

这将生成一个带有红色虚线的折线图

不同的线条

使用多个数据集可以创建多条线条，并在同一图形中显示它们。下面是一段代码，它将创建两个数据集，并在同一折线图中显示它们：

import matplotlib.pyplot as plt

df1 = pd.DataFrame({'x': range(10), 'y1': [1, 3, 2, 4, 5, 4, 6, 7, 8, 9]})
df2 = pd.DataFrame({'x': range(10), 'y2': [2, 1, 4, 2, 6, 4, 8, 6, 10, 9]})

plt.plot('x', 'y1', data=df1, color='blue')
plt.plot('x', 'y2', data=df2, color='red')
plt.title('Two Lines')
plt.xlabel('X')
plt.ylabel('Y')
plt.legend(['Line 1', 'Line 2'])
plt.show()

这将生成一个由两个线条组成的折线图

结论

在这篇文章中，我们介绍了如何使用Matplotlib为Pandas DataFrame绘制折线图。我们学习了如何使用plot()函数创建基本的折线图，如何添加标题和标签，如何改变线条样式和颜色，以及如何使用多个数据集在同一图形中显示多条线条。如今，您已经拥有了使用Matplotlib绘制Pandas DataFrame的折线图所需的知识和技能。