探索 Python：发现有趣的库——第 1 章：数据可视化之旅

在一个充满活力的科技世界中，数据分析专家“算法仙”和编程爱好者“代码侠”相遇了，决定一起踏上数据可视化的探险之旅。他们将运用 Matplotlib 和 Seaborn 这两个强大的 Python 库，将枯燥的数据转化为生动的图形。

算法仙：你好，代码侠！今天我们将一起探索数据可视化的奇妙世界。

代码侠：听起来很刺激！我们该从哪儿开始呢？

算法仙：首先，我们需要安装 seaborn 和 matplotlib。

 pip install seaborn matplotlib

基础图表

算法仙：接下来让我们从基础开始，用 Matplotlib 画一幅折线图如何？

代码侠：好的，折线图听起来像是一个很好的起点。

折线图

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.plot(x, y)
plt.xlabel('X Axis')
plt.ylabel('Y Axis')
plt.title('Simple Line Plot')
plt.show()

在这里插入图片描述

代码侠：哇，真是太有趣了！那柱状图呢？

算法仙：柱状图是展示和比较不同类别数据的绝佳方式。

柱状图

import matplotlib.pyplot as plt
import seaborn as sns

# 数据
data = {
    "categories": ['Category A', 'Category B', 'Category C'],
    "values":  [10, 20, 15]
}

# 绘制柱状图
sns.barplot(data=data, x='categories', y='values', hue='categories')
plt.title('Category Comparison')
plt.show()

在这里插入图片描述

代码侠：那散点图又是用来做什么的？

算法仙：散点图是探索两个变量之间关系的强大工具。

散点图

import matplotlib.pyplot as plt
import seaborn as sns

# 数据
data = {
    'X Value': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    'Y Value': [2, 4, 5, 4, 5, 7, 8, 6, 5, 4]
}

# 绘制散点图
sns.scatterplot(data=data, x='X Value', y='Y Value', hue='Y Value')
plt.title('Scatter Plot')
plt.show()

在这里插入图片描述

算法仙：看看这些点，它们似乎在告诉我们一些有趣的故事。

进阶可视化

代码侠：听说你最近在研究进阶的数据可视化技术？我也对这个很感兴趣。

算法仙：对啊，我正在探索如何使用 Seaborn 和 Matplotlib 制作更复杂的图表。你知道，不仅仅是普通的折线图和柱状图，而是更炫酷的那种。

代码侠：比如说？

算法仙：比如热力图、对比图和3D图表。这些可视化方式能帮助我们更好地理解和展示复杂的数据集。

代码侠：听起来很有趣！你能给我演示一下如何做一个热力图吗？

算法仙：让我给你看一个例子。比如我们有一个关于城市温度的数据集，我们可以用热力图来表示每个城市一周内的温度变化。

热力图

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

# 假设数据
data = {
    "Monday": [22, 24, 23, 22, 25, 24, 23],
    "Tuesday": [21, 23, 22, 21, 24, 23, 22],
    "Wednesday": [25, 26, 24, 25, 28, 27, 26],
    "Thursday": [23, 24, 23, 22, 25, 24, 23],
    "Friday": [20, 22, 21, 20, 23, 22, 21],
    "Saturday": [19, 21, 20, 19, 22, 21, 20],
    "Sunday": [18, 20, 19, 18, 21, 20, 19]
}

# 转换数据
temperature_df = pd.DataFrame(data)

# 绘制热力图
sns.heatmap(temperature_df, annot=True)
plt.title("Weekly City Temperature")
plt.show()

在这里插入图片描述
代码侠：哇！这图看起来真酷。

算法仙：是吧！热力图非常适合展示这类数据。接下来我们看看对比图, 对比图可以帮助我们比较两组数据之间的差异。

对比图

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

# 创建数据
data = {
    'Group A': np.random.randn(100),
    'Group B': np.random.randn(100) + 1
}

# 使用 Seaborn 绘制对比图
sns.histplot(data=data, kde=True)

# 设置坐标轴标签和标题
plt.xlabel('Value')
plt.ylabel('Frequency')
plt.title('Comparison of AB group data')

# 显示图形
plt.show()

在这里插入图片描述

代码侠：这让我更容易理解数据之间的不同关系了！每个图表都像讲述着一个独特的故事。

算法仙：现在，你想不想试试3D图表？虽然需要一点点额外的工作，但效果绝对值得。

代码侠：当然想！你知道我对这些可视化技术总是充满好奇。

算法仙：好的，那我们就用 Matplotlib 工具包来绘制一个3D散点图。比如说，我们有一些关于不同产品的销量、价格和顾客满意度的数据。

3D 图表

import matplotlib.pyplot as plt

# 产品数据
sales = [100, 200, 300, 400, 500]
prices = [20, 30, 40, 50, 60]
satisfaction = [3.5, 4.2, 2.8, 4.5, 4.0]

# 创建3D图表
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制3D散点图
ax.scatter(sales, prices, satisfaction, c='r', marker='o')
ax.set_xlabel('Sales')
ax.set_ylabel('Price')
ax.set_zlabel('Customer Satisfaction')

# 展示图表
plt.title('3D Scatter Plot')
plt.show()

在这里插入图片描述

代码侠：哇，这简直是艺术品！我们能从不同的角度看到数据。

算法仙：没错，3D 图表让我们的分析更加全面。

代码侠：太棒了，这些图表简直是数据可视化的魔法！

算法仙：确实如此。这些高级可视化技术不仅让数据分析更有效，还能让我们的报告更加吸引人。你也可以尝试一下，一旦开始，你就会发现有无限的可能性在等着你。

代码侠：我觉得我已经准备好挑战一个真实项目了。

算法仙：那正好！我们来分析一个真实的数据集，用我们刚学的技巧来可视化它。

实战项目：数据集可视化分析

代码侠和算法仙正在准备一个数据可视化的实战项目，计划使用一个现实数据集进行深入分析。

代码侠：算法仙，我听说我们今天要弄个大东西？

算法仙：没错，代码侠！我们今天要对一个真实的数据集进行可视化分析。

代码侠：哦！那我们要用什么数据集？

算法仙：我们来用著名的泰坦尼克号数据集。首先，我们需要安装 pandas 和 seaborn。

pip install pandas seaborn

算法仙：安装好了之后，我们可以开始导入数据集。

代码侠：就像打开宝箱一样，我迫不及待了！

import seaborn as sns

# 加载数据集
titanic = sns.load_dataset('titanic')

# 查看前几行数据
print(titanic.head())

在这里插入图片描述

算法仙：好的，数据到手了。现在，让我们来看看乘客的生存情况分布。

import seaborn as sns
import matplotlib.pyplot as plt


# 加载数据集
titanic = sns.load_dataset('titanic')

# 绘制生存情况的计数图
sns.countplot(x='survived', data=titanic)
plt.title('Titanic Survivors')
plt.show()

在这里插入图片描述
代码侠：这图表很直观，我们能从性别角度来看看生存情况吗？

算法仙：当然可以，来看看这个：

# 绘制性别和生存情况的对比图
import seaborn as sns
import matplotlib.pyplot as plt


# 加载数据集
titanic = sns.load_dataset('titanic')

# 绘制性别和生存情况的对比图
sns.countplot(x='survived', hue='sex', data=titanic)
plt.title('Survival by Gender on Titanic')
plt.show()

在这里插入图片描述

代码侠：哇，看来女士优先原则在泰坦尼克号上得到了体现。

算法仙：确实如此。现在，让我们深入一点，分析一下年龄对生存的影响。

import seaborn as sns
import matplotlib.pyplot as plt


# 加载数据集
titanic = sns.load_dataset('titanic')

# 年龄和生存情况的分布图
sns.histplot(data=titanic, x='age', hue='survived', kde=True, multiple="stack")
plt.title('Age Distribution of Survivors')
plt.show()