数据来源:https://www.kaggle.com/c/rsna-intracranial-hemorrhage-detection/data —— stage_2_train.csv

API

  • groupy
  • unstack
  • value_counts

csv data

在这里插入图片描述
查看 label

label = traindf.Label.values
label

array([0, 0, 0, …, 0, 0, 0])

分解ID 和subtype

traindf = traindf.ID.str.rsplit("_", n=1, expand=True)
traindf.head()

在这里插入图片描述
合并label

traindf.loc[:, "label"] = label
traindf.head()

在这里插入图片描述
0,1 列名重命名

traindf = traindf.rename({0: "id", 1: "subtype"}, axis=1)
traindf.head()

在这里插入图片描述

统计标签类别数

subtype_counts = traindf.groupby("subtype").label.value_counts()

在这里插入图片描述

unstack

subtype_counts.unstack()

在这里插入图片描述

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐