问题描述
最近在准备医学图像分割的模型训练,找到的数据集是用csv文件进行标注的,所以需要读取csv文件中的信息
关于如何将csv标注文件的标注信息与对应的图像联系起来,暂时还没开始
这里仅做一个预备的工作
读取操作
读取csv文件
import pandas as pd
# 修改自定义的路径
# header表示表头,即第0行的关于列的信息
# 取None值时无表头,自动添加数字作为列数
data = pd.read_csv('自定义的数据集路径或者链接',header)
对csv文件进行抽样
抽取行数据
# 其中n为int类型
data.sample(n) # 随机显示 n 条数据,默认n = 1
data.head(n) # 显示前 n 条数据,默认n = 5
data.tail(n) # 显示后 n 条数据,默认n = 5
抽取列数据
col_1 = data["列的表头名称"] # 获取一列数据
col_2 = data[["列的表头名称1","列的表头名称2"]]
# 获取两列及以上的信息,则需要使用二维的形式
抽取特定属性的数据
# 比如class列中的标注信息,0表示一个类别,需要将其提取出来
data_0 = data[data["class"] == 0]
处理csv标注文件与数据集
这里需要处理CT图像的MHD文件及其对应RAW文件,未完待续
备注参考
数据集与csv文件读取
pandas详细应用
python批量读取MHD与RAW文件
python读取MHD与RAW文件
未完待续。