一次学会Python数据可视化

一次学会Python数据可视化

更多文章代码详情可以查看博主个人网站：https://www.iwtmbtly.com/

---

​ 今天来讲讲 Python 的可视化技术。 如果你想要用 Python 进行数据分析，就需要在项目初期开始进行探索性的数据分析，这样 方便你对数据有一定的了解。其中最直观的就是采用数据可视化技术，这样，数据不仅一目 了然，而且更容易被解读。同样在数据分析得到结果之后，我们还需要用到可视化技术，把最终的结果呈现出来。

思维导图

视图分类

按照数据之间的关系，可以把可视化视图划分为 4 类，它们分别是比较、联系、构成和分布。我来简单介绍下这四种关系的特点：

• 比较：比较数据间各类别的关系，或者是它们随着时间的变化趋势，比如折线图；
• 联系：查看两个或两个以上变量之间的关系，比如散点图；
• 构成：每个部分占整体的百分比，或者是随着时间的百分比变化，比如饼状图；
• 分布：关注单个变量，或者多个变量的分布情况，比如直方图。

同样，按照变量的个数，我们可以把可视化视图划分为单变量分析和多变量分析。

单变量分析指的是一次只关注一个变量。比如我们只关注“身高”这个变量，来看身高的取 值分布，而暂时忽略其他变量。

多变量分析可以让你在一张图上可以查看两个以上变量的关系。比如“身高”和“年龄”， 你可以理解是同一个人的两个参数，这样在同一张图中可以看到每个人的“身高”和“年 龄”的取值，从而分析出来这两个变量之间是否存在某种联系。

可视化的视图可以说是分门别类，多种多样，今天我主要介绍常用的 10 种视图，这些视图 包括了散点图、折线图、直方图、条形图、箱线图、饼图、热力图、蜘蛛图、二元变量分布 和成对关系。

下面进行一一进行介绍。

工具包

在 Python 里包括了众多可视化库，比如 Matplotlib、Seaborn、Bokeh、Plotly、 Pyecharts、Mapbox 和 Geoplotlib。其中使用频率最高，最需要掌握的就是 Matplotlib 和 Seaborn。

Matplotlib 是 Python 的可视化基础库，作图风格和 MATLAB 类似，所以称为 Matplotlib。一般学习 Python 数据可视化，都会从 Matplotlib 入手，然后再学习其他的 Python 可视化库。

提前声明

下文中统一使用Matplotlib和Seaborn库进行画图操作，期间可能会遇到一些小问题。

1.MAtplotlib绘图显示中文问题

在画图前，使用一下代码即可解决中文显示问题：

import matplotlib
# 把默认文字设置为微软雅黑，粗体
matplotlib.rc("font",family='MicroSoft YaHei',weight="bold")

2.Seaborn数据集问题

下文画图演示中会使用到Seaborn的一些自带数据集，国内可能会加载失败，可以通过这里下载数据集到本地，再把数据集放在用户目录下即可。或者参考[这篇文章]((2条消息) 解决seaborn无法加载数据集错误_lymboy的博客-CSDN博客_数据集打开错误)，都可以得到解决。

常用视图

散点图

散点图的英文叫做 scatter plot，它将两个变量的值显示在二维坐标中，非常适合展示两个 变量之间的关系。当然，除了二维的散点图，我们还有三维的散点图。

在 Matplotlib 中，我们经常会用 到 pyplot 这个工具包，它包括了很多绘图函数，类似 Matlab 的绘图框架。在使用前你需要进行引用：

import matplotlib.pyplot as plt

在工具包引用后，画散点图，需要使用 plt.scatter(x, y, marker=None) 函数。x、y 是坐 标，marker 代表了标记的符号。比如“x”、“>”或者“o”。选择不同的 marker，呈 现出来的符号样式也会不同，你可以自己试一下。

除了 Matplotlib 外，你也可以使用 Seaborn 进行散点图的绘制。在使用 Seaborn 前，也 需要进行包引用：

import seaborn as sns

在引用 seaborn 工具包之后，就可以使用 seaborn 工具包的函数了。如果想要做散点图， 可以直接使用 sns.jointplot(x, y, data=None, kind='scatter') 函数。其中 x、y 是data 中的下标。data 就是我们要传入的数据，一般是 DataFrame 类型。kind 这类我们取 scatter，代表散点的意思。当然 kind 还可以取其他值，这个在后面的视图中会讲到，不同的 kind 代表不同的视图绘制方式。

好了，让我们来模拟下，假设我们的数据是随机的 1000 个点。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
N = 1000
x = np.random.randn(N)
y = np.random.randn(N)
# 用 Matplotlib 画散点图
# 用Matplotlib画散点图
plt.scatter(x, y, marker='x')	# 使用'x'标记符号
plt.show()
plt.scatter(x, y, marker='>')	# 使用'>'标记符号
plt.show()
plt.scatter(x, y, marker='o')	# 使用'o'标记符号
plt.show()
# 用 Seaborn 画散点图
df = pd.DataFrame({'x': x, 'y': y})
sns.jointplot(x="x", y="y", data=df, kind='scatter');
plt.show()

我们运行一下这个代码，就可以看到下面的视图（第1,2,3张图为 Matplotlib 绘制的，第4张 图为 Seaborn 绘制的）。其实你能看到 Matplotlib 和 Seaborn 的视图呈现还是有差别 的。Matplotlib 默认情况下呈现出来的是个长方形。而 Seaborn 呈现的是个正方形，而且 不仅显示出了散点图，还给了这两个变量的分布情况。

Matplotlib 绘制：

Seaborn 绘制：

折线图

折线图可以用来表示数据随着时间变化的趋势。

在 Matplotlib 中，我们可以直接使用 plt.plot() 函数，当然需要提前把数据按照 x 轴的大 小进行排序，要不画出来的折线图就无法按照 x 轴递增的顺序展示。

在 Seaborn 中，我们使用sns.lineplot (x, y, data=None)函数。其中 x、y 是 data 中的 下标。data 就是我们要传入的数据，一般是 DataFrame 类型。

这里我设置了 x、y 的数组。x 数组代表时间（年），y 数组随便设置几个取值。下面是详细的代码。

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
x = [2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019]
y = [5, 3, 6, 20, 17, 16, 19, 30, 32, 35]
# 使用matplotlib画折线图
plt.plot(x, y)
plt.show()
# 使用Seaborn画折线图
df = pd.DataFrame({'x':x, 'y':y})
sns.lineplot(x='x', y='y', data=df)
plt.show()

然后我们分别用 Matplotlib 和 Seaborn 进行画图，可以得到下面的图示。你可以看出这 两个图示的结果是完全一样的，只是在 seaborn 中标记了 x 和 y 轴的含义。

Matplotlib 绘制：

Seaborn 绘制：

直方图

直方图是比较常见的视图，它是把横坐标等分成了一定数量的小区间，这个小区间也叫 作“箱子”，然后在每个“箱子”内用矩形条（bars）展示该箱子的箱子数（也就是 y 值），这样就完成了对数据集的直方图分布的可视化。

在 Matplotlib 中，我们使用 plt.hist(x, bins=10) 函数，其中参数 x 是一维数组，bins 代 表直方图中的箱子数量，默认是 10。

在 Seaborn 中，我们使用 sns.distplot(x, bins=10, kde=True) 函数。其中参数 x 是一维 数组，bins 代表直方图中的箱子数量，kde 代表显示核密度估计，默认是 True，我们也可 以把 kde 设置为 False，不进行显示。核密度估计是通过核函数帮我们来估计概率密度的 方法。

这是一段绘制直方图的代码。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
a = np.random.randn(100)
s = pd.Series(a)
# 用matplotlib画直方图
plt.hist(s)
plt.show()
# 用seaborn画直方图
sns.distplot(s, kde=False)
plt.show()
sns.distplot(s, kde=True)
plt.show()

我们创建一个随机的一维数组，然后分别用 Matplotlib 和 Seaborn 进行直方图的显示， 结果如下，你可以看出，没有任何差别，其中最后一张图就是 kde 默认为 Ture 时的显示 情况。

条形图

如果说通过直方图可以看到变量的数值分布，那么条形图可以帮我们查看类别的特征。在条 形图中，长条形的长度表示类别的频数，宽度表示类别

在 Matplotlib 中，我们使用 plt.bar(x, height) 函数，其中参数 x 代表 x 轴的位置序列， height 是 y 轴的数值序列，也就是柱子的高度。

在 Seaborn 中，我们使用 sns.barplot(x=None, y=None, data=None) 函数。其中参数 data 为 DataFrame 类型，x、y 是 data 中的变量。

import matplotlib.pyplot as plt
import seaborn as sns

# 数据准备
x = ['Cat1', 'Cat2', 'Cat3', 'Cat4', 'Cat5']
y = [5, 4, 8, 12, 7]
# 用matplotlib画条形图
plt.bar(x, y)
plt.show()
# 用seaborn画条形图
sns.barplot(x, y)
plt.show()

我们创建了 x、y 两个数组，分别代表类别和类别的频数，然后用 Matplotlib 和 Seaborn 进行条形图的显示，结果如下：

箱线图

箱线图，又称盒式图，它是在 1977 年提出的，由五个数值点组成：最大值 (max)、最小值 (min)、中位数 (median) 和上下四分位数 (Q3, Q1)。它可以帮我们分析出数据的差异性、 离散程度和异常值等。

在 Matplotlib 中，我们使用 plt.boxplot(x, labels=None) 函数，其中参数 x 代表要绘制 箱线图的数据，labels 是缺省值，可以为箱线图添加标签。

在 Seaborn 中，我们使用 sns.boxplot(x=None, y=None, data=None) 函数。其中参数 data 为 DataFrame 类型，x、y 是 data 中的变量。

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

# 数据准备
# 生成 0-1 之间的 10*4 维度数据
data = np.random.normal(size=(10, 4))
lables = ['A', 'B', 'C', 'D']
# 用 Matplotlib 画箱线图
plt.boxplot(data, labels=lables)
plt.show()
# 用 Seaborn 画箱线图
df = pd.DataFrame(data, columns=lables)
sns.boxplot(data=df)
plt.show()

这段代码中，我生成 0-1 之间的 10*4 维度数据，然后分别用 Matplotlib 和 Seaborn 进 行箱线图的展示，结果如下：

饼状图

饼图是常用的统计学模块，可以显示每个部分大小与总和之间的比例。在 Python 数据可视 化中，它用的不算多。我们主要采用 Matplotlib 的 pie 函数实现它。

在 Matplotlib 中，我们使用 plt.pie(x, labels=None) 函数，其中参数 x 代表要绘制饼图 的数据，labels 是缺省值，可以为饼图添加标签。

这里我设置了lables数组，分别代表高中、本科、硕士、博士和其他几种学历的分类标 签。nums 代表这些学历对应的人数。

import matplotlib.pyplot as plt

# 数据准备
nums = [25, 37, 33, 37, 8]
labels = ['高中生', '本科生', '硕士生', '博士生', '其他']
# 用 Matplotlib 画饼图
plt.pie(x=nums, labels=labels, autopct= '%1.1f%%')
plt.title("--饼状图--")
plt.show()

通过 Matplotlib 的 pie 函数，我们可以得出下面的饼图：

热力图

热力图，英文叫 heat map，是一种矩阵表示方法，其中矩阵中的元素值用颜色来代表，不 同的颜色代表不同大小的值。通过颜色就能直观地知道某个位置上数值的大小。另外你也可 以将这个位置上的颜色，与数据集中的其他位置颜色进行比较。

热力图是一种非常直观的多元变量分析方法。

我们一般使用 Seaborn 中的 sns.heatmap(data) 函数，其中 data 代表需要绘制的热力图数据。

这里我们使用 Seaborn 中自带的数据集 flights，该数据集记录了 1949 年到 1960 年期 间，每个月的航班乘客的数量。

import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
flights = sns.load_dataset("flights")
data=flights.pivot('year','month','passengers')
# 用 Seaborn 画热力图
sns.heatmap(data)
plt.show()

通过 seaborn 的 heatmap 函数，我们可以观察到不同年份，不同月份的乘客数量变化情 况，其中颜色越浅的代表乘客数量越多，如下图所示：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-oFhxaBV8-1655801366676)(C:\Users\liuya\AppData\Roaming\Typora\typora-user-images\image-20220621161150364.png)]

雷达图

蜘蛛图是一种显示一对多关系的方法。在蜘蛛图中，一个变量相对于另一个变量的显著性是 清晰可见的。

假设我们想要给王者荣耀的玩家做一个战力图，指标一共包括推进、KDA、生存、团战、 发育和输出。那该如何做呢？

这里我们需要使用 Matplotlib 来进行画图，首先设置两个数组：labels 和 stats。他们分 别保存了这些属性的名称和属性值。 因为雷达图是一个圆形，你需要计算每个坐标的角度，然后对这些数值进行设置。当画完最 后一个点后，需要与第一个点进行连线。

因为需要计算角度，所以我们要准备 angles 数组；又因为需要设定统计结果的数值，所以 我们要设定 stats 数组。并且需要在原有 angles 和 stats 和labels数组上增加一位，也就是添加数组的第一个元素。

import numpy as np
import matplotlib.pyplot as plt

# 数据准备
labels = np.array(["推进", "KDA", "生存", "团战", "发育", "输出"])
stats = [83, 61, 95, 67, 76, 88]
# 画图数据准备，角度值，状态值
angles = np.linspace(0, 2 * np.pi, len(labels), endpoint=False)
stats = np.concatenate((stats, [stats[0]]))
angles = np.concatenate((angles, [angles[0]]))
labels = np.concatenate((labels, [labels[0]]))  #对labels进行封闭
# 用matplotlib画蜘蛛图
fig = plt.figure()
ax = fig.add_subplot(111, polar=True)
ax.plot(angles, stats, 'o-', linewidth=2)
ax.fill(angles, stats, alpha=0.25)
ax.set_thetagrids(angles * 180 / np.pi, labels)
plt.show()

代码中 flt.figure() 是创建一个空白的 figure 对象，这样做的目的相当于画画前先准备一个 空白的画板。然后 add_subplot(111) 可以把画板划分成 1 行 1 列。再用 ax.plot 和 ax.fill 进行连线以及给图形上色。最后我们可以得到下面的雷达图，看起来是不是很酷？

二元变量分布

如果我们想要看两个变量之间的关系，就需要用到二元变量分布。当然二元变量分布有多种 呈现方式，开头给你介绍的散点图就是一种二元变量分布。

在 Seaborn 里，使用二元变量分布是非常方便的，直接使用 sns.jointplot(x, y, data=None, kind) 函数即可。其中用 kind 表示不同的视图类 型：“kind=‘scatter’”代表散点图，“kind=‘kde’”代表核密度 图，“kind=‘hex’ ”代表 Hexbin 图，它代表的是直方图的二维模拟。

这里我们使用 Seaborn 中自带的数据集 tips，这个数据集记录了不同顾客在餐厅的消费账 单及小费情况。代码中 total_bill 保存了客户的账单金额，tip 是该客户给出的小费金额。 我们可以用 Seaborn 中的 jointplot 来探索这两个变量之间的关系。

import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
tips = sns.load_dataset("tips")
print(tips.head(10))
# 用 Seaborn 画二元变量分布图（散点图，核密度图，Hexbin 图）
sns.jointplot(x="total_bill", y="tip", data=tips, kind='scatter')
sns.jointplot(x="total_bill", y="tip", data=tips, kind='kde')
sns.jointplot(x="total_bill", y="tip", data=tips, kind='hex')
plt.show()

代码中我用 kind 分别显示了他们的散点图、核密度图和 Hexbin 图，如下图所示：

成对关系

如果想要探索数据集中的多个成对双变量的分布，可以直接采用 sns.pairplot() 函数。它会 同时展示出 DataFrame 中每对变量的关系，另外在对角线上，你能看到每个变量自身作为 单变量的分布情况。它可以说是探索性分析中的常用函数，可以很快帮我们理解变量对之间 的关系。

pairplot() 函数的使用，就像在 DataFrame 中使用 describe() 函数一样方便，是数据探索 中的常用函数。

这里我们使用 Seaborn 中自带的 iris 数据集，这个数据集也叫鸢尾花数据集。鸢尾花可以 分成 Setosa、Versicolour 和 Virginica 三个品种，在这个数据集中，针对每一个品种，都 有 50 个数据，每个数据中包括了 4 个属性，分别是花萼长度、花萼宽度、花瓣长度和花瓣 宽度。通过这些数据，需要你来预测鸢尾花卉属于三个品种中的哪一种。

import matplotlib.pyplot as plt
import seaborn as sns
# 数据准备
iris = sns.load_dataset('iris')
# 用 Seaborn 画成对关系
sns.pairplot(iris)
plt.show()

这里我们用 Seaborn 中的 pairplot() 函数来对数据集中的多个双变量的关系进行探索，如下 图所示。从图上你能看出，一共有 sepal_length、sepal_width、petal_length 和 petal_width4 个变量，它们分别是花萼长度、花萼宽度、花瓣长度和花瓣宽度。

下面这张图相当于这 4 个变量两两之间的关系。比如矩阵中的第一张图代表的就是花萼长 度自身的分布图，它右侧的这张图代表的是花萼长度与花萼宽度这两个变量之间的关系。

总结

以上是对 Python 可视化工具包 Matplotlib 和 Seaborn 工具包的使用。他们两者之间的关系就相当于 NumPy 和 Pandas 的关系。Seaborn 是基于 Matplotlib 更加高级的 可视化库。

另外针对这 10 种可视化视图，可以按照变量之间的关系对它们进行分类，这些关系分别是比较、联系、构成和分布。当然我们也可以按照随机变量的个数来进行划分，比如 单变量分析和多变量分析。在数据探索中，成对关系 pairplot() 的使用，相好比 Pandas 中的 describe() 使用一样方便，常用于项目初期的数据可视化探索。

在 Matplotlib 和 Seaborn 的函数中，我只列了最基础的使用，方便快速上手。当然如果你也可以设置修改颜色、宽度等视图属性。你可以自己查看相关的函数帮助文档。这些留给你来进行探索。