pandas为气泡图添加中文文本标签|极客教程

pandas为气泡图添加中文文本标签

引言

在数据可视化中，气泡图是一种常见且有趣的图表类型。它能够同时展示两个维度的数据，通过气泡的大小和颜色来反映第三个维度的数据，能够更直观地帮助我们分析和理解数据的规律。然而，在使用气泡图时，往往需要为气泡添加标签，以更清晰地表达数据。本文将详细介绍如何使用Python的pandas库为气泡图添加中文文本标签。

1. 准备工作

首先，我们需要确保已经安装了pandas库。如果尚未安装，可以使用以下命令进行安装：

pip install pandas

同时，我们还需要一些示例数据。本文将使用一个包含了城市、人口和GDP数据的DataFrame。示例数据如下：

import pandas as pd

# 创建示例数据
data = {
    'City': ['北京', '上海', '广州', '深圳', '天津', '重庆'],
    'Population': [2171, 2424, 1461, 1303, 1562, 3016],
    'GDP': [30319, 32680, 23678, 27226, 30322, 19424]
}

df = pd.DataFrame(data)

2. 气泡图绘制

使用pandas的plot方法可以很方便地绘制气泡图。我们可以指定x和y轴的数据，以及气泡的大小和颜色。在这里，我们将城市的人口作为y轴，GDP作为x轴，人口大小作为气泡的大小，GDP的大小作为气泡的颜色。

# 绘制气泡图
df.plot.scatter(x='GDP', y='Population', s=df['Population']/100, c=df['GDP']/1000, colormap='viridis')

通过上述代码，我们可以得到一个基本的气泡图

3. 添加标签

为了使气泡图更加清晰和易读，我们经常需要为气泡添加标签，特别是对于气泡的中心位置。Pandas库提供了很多方法来添加标签，其中一种方法是使用annotate函数。

# 为气泡图添加标签
for i, row in df.iterrows():
    plt.annotate(row['City'], (row['GDP'], row['Population']), fontsize=8, ha='center')

在上述代码中，我们使用了iterrows方法来遍历DataFrame的每一行。然后，我们使用annotate函数将每个城市的名称标签添加到气泡的中心位置，参数fontsize用于指定字体大小，ha用于指定水平对齐方式。

运行以上代码，我们可以得到一个带有中文文本标签的气泡图

4. 支持中文文本

在默认情况下，matplotlib并不支持中文文本的显示，需要进行一些额外的设置。我们需要下载中文字体文件，并将其设置为matplotlib的默认字体。以下是一种常用的设置方法：

import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties

# 设置中文字体
font = FontProperties(fname='/path/to/simhei.ttf')
plt.rcParams['font.family'] = font.get_name()

其中，fname参数指定了中文字体文件的路径，可以根据实际情况进行调整。

设置完中文字体后，我们再次运行添加标签的代码，就能够正确显示中文文本了。