Python Pandas – 使用Seaborn创建计数图并为条形图设置样式

在数据分析中，计数图是一种非常有用的可视化工具，可以用来展示分类变量的数量。而Python中的Pandas和Seaborn库可以方便地创建各种类型的计数图并设置样式。

更多Pandas相关文章，请阅读：Pandas 教程

安装Pandas和Seaborn库

在使用Pandas和Seaborn之前，我们需要先安装它们。可以通过以下命令在终端（Mac或Linux系统）或命令提示符（Windows系统）中进行安装：

pip install pandas seaborn

创建数据集

在本文中，我们将使用一个虚构的数据集来演示计数图的创建和样式设置。该数据集包含了2019年世界杯足球赛32支参赛队伍的数据，包括国家名称、比赛胜利和失败次数等。数据集如下所示：

import pandas as pd

# 创建数据集
data = {'Country': ['Russia', 'Germany', 'Brazil', 'Portugal', 'Argentina', 'Belgium', 'Poland', 'France', 'Spain',
                    'Peru', 'Switzerland', 'England', 'Colombia', 'Mexico', 'Uruguay', 'Croatia', 'Denmark',
                    'Iceland', 'Costa Rica', 'Sweden', 'Tunisia', 'Egypt', 'Senegal', 'Iran', 'Serbia',
                    'Nigeria', 'Japan', 'South Korea', 'Saudi Arabia', 'Morocco', 'Australia', 'Panama'],
        'Wins': [2, 1, 3, 1, 1, 3, 1, 7, 3, 2, 1, 3, 2, 2, 4, 2, 1, 1, 0, 3, 1, 0, 2, 1, 1, 1, 1, 0, 1, 0, 0, 0],
        'Losses': [1, 0, 1, 1, 1, 0, 2, 2, 2, 1, 0, 1, 1, 1, 1, 1, 1, 2, 2, 1, 2, 2, 1, 2, 1, 2, 2, 2, 2, 1, 2, 1]}

df = pd.DataFrame(data)
df.head()

输出结果为：

     Country  Wins  Losses
0     Russia     2       1
1    Germany     1       0
2      Brazil    3       1
3   Portugal     1       1
4  Argentina     1       1

创建基本计数图

我们可以使用Seaborn的countplot()函数创建基本计数图。该函数接受一个Pandas数据框（DataFrame）作为输入，并基于给定的分类变量绘制计数图。

下面的示例代码展示了如何使用countplot()函数创建一个基本计数图：

import seaborn as sns

# 创建基本计数图
sns.countplot(x='Wins', data=df)

设置计数图的样式

Seaborn库提供了各种函数来设置计数图的样式。下面是一些常用的样式设置函数：

set_style() – 设置绘图风格。
set_palette() – 设置颜色。
set_context() – 设置绘图元素大小的缩放比例。

下面的示例代码展示了如何设置计数图的颜色和绘图风格：

# 设置计数图颜色和绘图风格
sns.set_palette('pastel')
sns.set_style('whitegrid')

# 创建计数图
sns.countplot(x='Wins', data=df)

更改计数图的柱状条颜色和顺序

在某些情况下，我们可能需要自定义计数图中的柱状条颜色和顺序。Seaborn提供了相应的函数来实现这些功能：palette和order。

palette – 用于设置颜色调色板，可以是字符串（例如“Blues”），也可以是颜色列表（例如[‘red’, ‘green’]）。
order – 用于指定条形的顺序。

下面是示例代码演示如何使用这些函数：

# 自定义颜色和顺序
custom_palette = sns.color_palette('pastel', 3)
order = [3, 2, 1]

# 创建计数图
sns.countplot(x='Wins', data=df, palette=custom_palette, order=order)

添加标签和标题

最后，我们可以添加标签和标题以进一步说明我们的计数图。xlabel、ylabel和title函数用于添加标签和标题。

下面是示例代码展示如何添加标签和标题：

# 创建计数图并添加标签和标题
sns.countplot(x='Wins', data=df, palette=custom_palette, order=order)
sns.set_context('talk')
sns.set_style('ticks')
plt.xlabel('Number of Wins')
plt.ylabel('Number of Teams')
plt.title('2019 World Cup Team Results')
plt.show()