如何用海底Python制作山脊线剧情?
先决条件: 海鸟
脊线图是一组重叠的密度图,有助于比较数据集之间的多种分布。山脊线图看起来像一个山脉,它们对于可视化分布随时间或空间的变化非常有用。有时也被称为“joyplot”,指的是 Joy Division 专辑《未知的快乐》的标志性封面艺术。在本文中,我们将看到如何为数据集生成脊线图。
装置
像任何其他 python 库一样,seaborn 可以使用 pip 轻松安装:
pip install seaborn
该库是 Anaconda 发行版的一部分,如果您的 IDE 受 Anaconda 支持,通常只需导入即可,但也可以通过以下命令安装:
conda install seaborn
程序
- 用 Python 加载生成脊线图所需的包。
- 读取数据集。在本例中,我们使用 read_csv() 方法加载数据集。在给定的示例中,我们将使用 head() 方法仅显示前 5 个条目。
- 生成 RidgePlot。脊线图使用刻面,这意味着它在一列中创建小倍数。要生成脊线图,Seaborn 使用 FacetGrid() 方法,所有需要的信息都应该传递给它
语法:海鸟。FacetGrid(数据、行、列、色调、调色板、纵横比、高度)
参数:
- 数据:整齐的(“长格式”)数据帧,其中每一列是一个变量,每一行是一个观察值。
- 行、列、色调:定义数据子集的变量,这些数据将绘制在网格中的独立面上。
- 高度:每个小平面的高度(英寸)。
- 纵横比:每个小平面的纵横比,因此纵横比*高度给出了每个小平面的宽度,单位为英寸。
- 调色板:用于色调变量不同级别的颜色。
样本数据库:以下示例中使用的数据集是从 kaggle.com 下载的。以下链接可用于相同目的。
数据库: titanic_train.csv
示例:
Python 3
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
from sklearn import preprocessing
df = pd.read_csv("titanic_train.csv")
df.dropna()
le = preprocessing.LabelEncoder()
df["Sex"] = le.fit_transform(df["Sex"])
rp = sns.FacetGrid(df, row="Sex", hue="Sex", aspect=5, height=1.25)
rp.map(sns.kdeplot, 'Survived', clip_on=False,
shade=True, alpha=0.7, lw=4, bw=.2)
rp.map(plt.axhline, y=0, lw=4, clip_on=False)
输出:
版权属于:月萌API www.moonapi.com,转载请注明出处