Seaborn可视化:图形个性化设置的几个小技巧

阅读量:     Authors: 阳哥 PYTHONVISUALIZATION
Seaborn


本文首发于我的微信公众号(ID:PyDataLab)。

1 概述

在可视化过程中,经常会对默认的制图效果不满意,希望能个性化进行各种设置。

本文通过一个简单的示例,来介绍seaborn可视化过程中的个性化设置。包括常用的设置,如:

  1. 设置图表显示颜色
  2. 设置图表标题,包括显示位置,字体大小,颜色等
  3. 设置x轴和y轴标题,包括颜色,字体大小
  4. 设置x轴和y轴刻度内容,包括颜色、字体大小、字体方向等
  5. 将x轴和y轴内容逆序显示
  6. 设置x轴或y轴显示位置

本文的运行环境:

  1. windows 7
  2. python 3.5
  3. jupyter notebook
  4. seaborn 0.7.1
  5. matplotlib 2.0.2

2 未个性化设置的情形

本文的数据来自UCI的数据集”sonar”,用pandas直接读取数据。如下:

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

target_url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/undocumented/connectionist-bench/sonar/sonar.all-data'
df = pd.read_csv(target_url, header=None, prefix='V')
corr = df.corr()

首先来看看没有进行个性化设置时的显示情况,如下:

f, ax= plt.subplots(figsize = (14, 10))

sns.heatmap(corr,cmap='RdBu', linewidths = 0.05, ax = ax)

# 设置Axes的标题
ax.set_title('Correlation between features')

f.savefig('sns_style_origin.jpg', dpi=100, bbox_inches='tight')

图片显示效果如下:


seaborn制图的默认效果其实还是不错的。

3 进行个性化设置

对于上面这张图,可能让y轴从下到上,从v0开始显示,这样显示出来的对角线可能更符合我们的视觉显示效果。

这就要用到 将y轴内容进行可逆显示,涉及的代码如下:

# 将y轴或x轴进行逆序
ax.invert_yaxis()
# ax.invert_xaxis()

其他的个性化设置的代码,包括:

将x轴刻度放置在top位置的几种方法

# 将x轴刻度放置在top位置的几种方法
# ax.xaxis.set_ticks_position('top')
ax.xaxis.tick_top()
# ax.tick_params(axis='x',labelsize=6, colors='b', labeltop=True, labelbottom=False) # x轴

设置坐标轴刻度参数,”axis”不写的时候,默认是x轴和y轴的参数同时调整。

# 设置坐标轴刻度的字体大小
# matplotlib.axes.Axes.tick_params
ax.tick_params(axis='y',labelsize=8) # y轴

旋转轴刻度上文字方向的两种方法

# 旋转轴刻度上文字方向的两种方法
ax.set_xticklabels(ax.get_xticklabels(), rotation=-90)
# ax.set_xticklabels(corr.index, rotation=90)

保存图片,设置bbox_inches=’tight’,保存的图片则不会出现部分内容显示不全的现象。

f.savefig('sns_style_update.jpg', dpi=100, bbox_inches='tight')

整合好的代码如下,大家可以运行试试效果。

f, ax = plt.subplots(figsize = (14, 10))

# 设置颜色
cmap = sns.cubehelix_palette(start = 1, rot = 3, gamma=0.8, as_cmap = True)

# color: https://matplotlib.org/users/colormaps.html
sns.heatmap(corr,cmap='RdBu', linewidths = 0.05, ax = ax)

# 设置Axes的标题
ax.set_title('Correlation between features', fontsize=18, position=(0.5,1.05))

# 将y轴或x轴进行逆序
ax.invert_yaxis()
# ax.invert_xaxis()

ax.set_xlabel('X Label',fontsize=10)

# 设置Y轴标签的字体大小和字体颜色
ax.set_ylabel('Y Label',fontsize=15, color='r')

# 设置坐标轴刻度的字体大小
# matplotlib.axes.Axes.tick_params
ax.tick_params(axis='y',labelsize=8) # y轴
# ax.tick_params(axis='x',labelsize=6, colors='b', labeltop=True, labelbottom=False) # x轴

# 将x轴刻度放置在top位置的几种方法
# ax.xaxis.set_ticks_position('top')
ax.xaxis.tick_top()
# ax.tick_params(axis='x',labelsize=6, colors='b', labeltop=True, labelbottom=False) # x轴

# 修改tick的字体颜色
# ax.tick_params(axis='x', colors='b') # x轴

# 旋转轴刻度上文字方向的两种方法
ax.set_xticklabels(ax.get_xticklabels(), rotation=-90)
# ax.set_xticklabels(corr.index, rotation=90)

# 单独设置y轴或x轴刻度的字体大小, 调整字体方向
# ax.set_yticklabels(ax.get_yticklabels(),fontsize=6)
# ax.set_xticklabels(ax.get_xticklabels(), rotation=-90)

f.savefig('sns_style_update.jpg', dpi=100, bbox_inches='tight')

图形显示效果如下:


这些个性化的设置,其实大部分都是使用的matplotlib的内容,seaborn是基于matplotlib衍生的,所以可以跟matplotlib进行融合使用。

当然,并不是每次都需要进行个性定制,具体可以根据自己的需求来设置。


对我的文章感兴趣的朋友,可以关注我的微信公众号「Python数据之道」(ID:PyDataLab),接收我的更新通知。

« Previous Next »