【Python入门系列】第十五篇:Python数据可视化和图表绘制


前言

数据可视化是数据分析和数据科学中非常重要的一部分。通过可视化,我们可以更好地理解数据、发现数据之间的关系、展示数据的趋势和模式,并向他人传达我们的发现。

Python是一种功能强大的编程语言,拥有许多用于数据可视化的库和工具。其中,Matplotlib是最常用的绘图库之一,它提供了各种绘图函数和方法,可以绘制折线图、散点图、柱状图、饼图等等。Seaborn是基于Matplotlib的高级绘图库,它提供了更美观和简化的绘图接口,能够轻松绘制统计图表。Plotly是一个交互式绘图库,可以创建漂亮的可交互图表和可视化仪表板。

在数据可视化过程中,我们通常需要先加载数据,然后选择合适的图表类型进行绘制,并对图表进行进一步的定制和美化。最后,我们可以使用适当的标题、标签和图例来解释和传达图表的含义。

一、可视化与绘图常用库

Python中常用的数据可视化和图表绘制库有以下几个:

  1. Matplotlib:Matplotlib是Python中最常用的绘图库之一。它提供了广泛的绘图功能,包括折线图、散点图、柱状图、饼图等。Matplotlib的接口灵活,可以进行各种定制和美化操作。

  2. Seaborn:Seaborn是基于Matplotlib的高级绘图库。它提供了更美观和简化的绘图接口,并且支持许多统计图表的绘制,如箱线图、热力图、小提琴图等。Seaborn还可以轻松地处理缺失值和异常值。

  3. Plotly:Plotly是一个交互式的绘图库,可以创建漂亮的可交互图表和可视化仪表板。它支持多种图表类型,包括折线图、散点图、柱状图、饼图等。Plotly还提供了在线共享和协作的功能。

二、Matplotlib

Matplotlib是Python中最常用的绘图库之一。它提供了广泛的绘图功能,包括折线图、散点图、柱状图、饼图等。

1、折线图

下面是一个简单的Matplotlib绘图示例,以折线图为例:

import matplotlib.pyplot as plt

# 创建数据
x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

# 绘制折线图
plt.plot(x, y)

# 添加标题和标签
plt.title('折线图示例')
plt.xlabel('x轴')
plt.ylabel('y轴')

# 显示图表
plt.show()

在这个示例中,我们首先导入 matplotlib.pyplot 模块,并创建了两个列表 x 和 y 作为数据。然后,使用 plt.plot() 函数绘制了折线图。接下来,我们使用 plt.title() 、 plt.xlabel() 和 plt.ylabel() 函数添加了标题和标签。最后,使用 plt.show() 函数显示了图表。

2、散点图

Matplotlib绘制散点图:

import matplotlib.pyplot as plt

# 创建数据
x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

# 绘制散点图
plt.scatter(x, y)

# 添加标题和标签
plt.title('散点图示例')
plt.xlabel('x轴')
plt.ylabel('y轴')

# 显示图表
plt.show()

在这个示例中,我们使用 plt.scatter() 函数绘制了散点图。其他部分的代码和之前的折线图示例相似。

3、柱状图:

Matplotlib绘制柱状图:

import matplotlib.pyplot as plt

# 创建数据
x = ['A', 'B', 'C', 'D', 'E']
y = [10, 15, 7, 12, 9]

# 绘制柱状图
plt.bar(x, y)

# 添加标题和标签
plt.title('柱状图示例')
plt.xlabel('类别')
plt.ylabel('数量')

# 显示图表
plt.show()

在这个示例中,我们使用 plt.bar() 函数绘制了柱状图。x轴的标签是一个字符串列表,y轴是对应的数值列表。

三、Seaborn

Seaborn是一个基于Matplotlib的Python数据可视化库,它提供了更高级的绘图接口和更美观的默认样式。Seaborn旨在简化数据可视化的过程,并且支持许多统计图表的绘制。

Seaborn相比于Matplotlib,具有以下几个优点:

  1. 更美观的默认样式:Seaborn提供了一些美观的默认样式,使得绘制的图表更加吸引人。
  2. 更简单的绘图接口:Seaborn的绘图接口更加简单,可以轻松地绘制各种图表,如箱线图、小提琴图、热力图等。
  3. 支持统计图表:Seaborn内置了许多统计图表的绘制函数,可以直接绘制例如分布图、回归图、分类图等常见的统计图表。
  4. 与Pandas集成:Seaborn可以与Pandas无缝集成,可以直接从Pandas的数据框中绘制图表。

1、散点图

面是一个简单的Seaborn绘图示例,以绘制散点图为例:

import seaborn as sns
import matplotlib.pyplot as plt

# 创建数据
tips = sns.load_dataset("tips")

# 绘制散点图
sns.scatterplot(data=tips, x="total_bill", y="tip")

# 添加标题和标签
plt.title('散点图示例')
plt.xlabel('总账单')
plt.ylabel('小费')

# 显示图表
plt.show()

在这个示例中,我们首先导入了Seaborn和Matplotlib的库。然后使用 sns.load_dataset() 函数加载了一个示例数据集(tips)。接下来,使用 sns.scatterplot() 函数绘制了散点图,其中 data 参数指定数据集, x 和 y 参数指定x轴和y轴的变量名。

2、箱线图

以下为Seaborn绘制箱线图:

import seaborn as sns

# 加载示例数据集
tips = sns.load_dataset("tips")

# 绘制箱线图
sns.boxplot(data=tips, x="day", y="total_bill")

# 添加标题和标签
plt.title('箱线图示例')
plt.xlabel('星期')
plt.ylabel('总账单')

# 显示图表
plt.show()

在这个示例中,我们使用 sns.boxplot() 函数绘制了箱线图。 data 参数指定了数据集, x 和 y 参数分别指定了x轴和y轴的变量名。

3、小提琴图

以下为使用Seaborn绘制小提琴图:

import seaborn as sns

# 加载示例数据集
tips = sns.load_dataset("tips")

# 绘制小提琴图
sns.violinplot(data=tips, x="day", y="total_bill")

# 添加标题和标签
plt.title('小提琴图示例')
plt.xlabel('星期')
plt.ylabel('总账单')

# 显示图表
plt.show()

这个示例中,我们使用 sns.violinplot() 函数绘制了小提琴图。其他部分的代码和之前的示例类似。

4、热力图

以下为使用Seaborn绘制热力图:

import seaborn as sns

# 加载示例数据集
flights = sns.load_dataset("flights")

# 将数据转换为矩阵形式
flights_matrix = flights.pivot("month", "year", "passengers")

# 绘制热力图
sns.heatmap(flights_matrix, annot=True, cmap="YlGnBu")

# 添加标题和标签
plt.title('热力图示例')
plt.xlabel('年份')
plt.ylabel('月份')

# 显示图表
plt.show()

在这个示例中,我们使用 sns.heatmap() 函数绘制了热力图。 annot=True 参数用于在每个单元格中显示数值, cmap 参数指定了颜色映射。

四、Plotly

Plotly是一个交互式的绘图库,可以创建漂亮的可交互图表和可视化仪表板。它支持多种图表类型,包括折线图、散点图、柱状图、饼图等。Plotly还提供了在线共享和协作的功能。

1、折线图

import plotly.graph_objects as go

# 创建数据
x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

# 创建折线图
fig = go.Figure(data=go.Scatter(x=x, y=y))

# 添加标题和轴标签
fig.update_layout(title='折线图示例', xaxis_title='x轴', yaxis_title='y轴')

# 显示图表
fig.show()

在这个示例中,我们使用 go.Scatter() 函数创建了一个折线图,并使用 go.Figure() 函数将其包装成一个图表对象。通过 update_layout() 函数可以添加标题和轴标签。

2、散点图

import plotly.graph_objects as go

# 创建数据
x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

# 创建散点图
fig = go.Figure(data=go.Scatter(x=x, y=y, mode='markers'))

# 添加标题和轴标签
fig.update_layout(title='散点图示例', xaxis_title='x轴', yaxis_title='y轴')

# 显示图表
fig.show()

在这个示例中,我们使用 mode=‘markers’ 参数将折线图转换为散点图。

3、条形图

import plotly.graph_objects as go

# 创建数据
x = ['A', 'B', 'C', 'D', 'E']
y = [10, 15, 7, 12, 9]

# 创建条形图
fig = go.Figure(data=go.Bar(x=x, y=y))

# 添加标题和轴标签
fig.update_layout(title='条形图示例', xaxis_title='类别', yaxis_title='数量')

# 显示图表
fig.show()

在这个示例中,我们使用 go.Bar() 函数创建了一个条形图。

总结

总之,Python数据可视化和图表绘制是通过使用Matplotlib和Plotly等库,将数据转化为可视化图表的过程。这些工具提供了丰富的功能和灵活性,可以满足各种数据可视化的需求。

猜你喜欢

转载自blog.csdn.net/qq_38628970/article/details/131579385