Python气泡图|极客教程

Python气泡图

介绍

气泡图是一种可视化工具，用于展示两个维度的数据，每个数据点由两个数值表示，其中一个数值决定了气泡的大小，另一个决定了气泡的位置。气泡图可以直观地呈现数据点的分布和关系，常用于分析和比较多个数据集。

Python作为一门强大的编程语言，拥有丰富的数据处理和可视化库。在本文中，我们将介绍如何使用Python绘制气泡图。

数据准备

首先，我们需要准备数据用于绘制气泡图。这里以一个电商平台的销售数据为例，假设我们有以下几个商品的销售情况：

# 商品名称
products = ['A', 'B', 'C', 'D', 'E']

# 销售额（单位：万元）
sales = [15, 20, 30, 10, 25]

# 市场份额（单位：%）
market_shares = [10, 15, 20, 8, 12]

# 广告投入（单位：万元）
advertising_costs = [5, 8, 12, 3, 6]

以上数据分别表示了每个商品的销售额、市场份额和广告投入。我们将使用这些数据绘制气泡图。

绘制气泡图

在Python中，我们可以使用matplotlib库来绘制气泡图。首先，我们需要导入相关的库和模块：

import matplotlib.pyplot as plt
import numpy as np

接下来，我们可以创建一个figure对象和相应的axes对象，然后使用scatter函数绘制气泡图：

# 创建figure对象和axes对象
fig, ax = plt.subplots()

# 绘制气泡图
ax.scatter(market_shares, advertising_costs, s=sales*100, alpha=0.7)

在上述代码中，s参数用于设置气泡的大小，这里我们将销售额放大100倍以便更好地展示。alpha参数用于设置气泡的透明度，取值范围为0到1。

接下来，我们可以对气泡图进行一些自定义的操作，比如添加标题、坐标轴标签和刻度等：

# 添加标题和坐标轴标签
ax.set_title('Sales Analysis')
ax.set_xlabel('Market Shares (%)')
ax.set_ylabel('Advertising Costs (万元)')

# 设置x轴和y轴的刻度范围
ax.set_xlim([0, max(market_shares) * 1.2])
ax.set_ylim([0, max(advertising_costs) * 1.2])

# 设置x轴和y轴的刻度
ax.set_xticks(np.arange(0, max(market_shares) * 1.2, 5))
ax.set_yticks(np.arange(0, max(advertising_costs) * 1.2, 2))

# 添加数据标签
for i, product in enumerate(products):
    ax.text(market_shares[i], advertising_costs[i], product, ha='center')

# 显示图形
plt.show()

在上述代码中，我们使用set_title、set_xlabel和set_ylabel函数分别添加了标题和坐标轴标签。set_xlim和set_ylim函数用于设置x轴和y轴的刻度范围，set_xticks和set_yticks函数用于设置刻度。最后，我们使用text函数在每个气泡上添加了对应的商品名称。

运行以上代码，将会得到一个销售分析的气泡图