在 Python 中将数据加载到训练集中答案

【问题标题】：Loading data into training set in Python在 Python 中将数据加载到训练集中
【发布时间】：2020-02-11 22:40:44
【问题描述】：

我需要为我的 2 个目录中的所有文件创建一个训练和测试集。但我只能选择一个文件。如何浏览所有的 .npz 文件？

图像存储在图像文件夹中，格式为：[patient_id].npz。每个 .npz 文件都包含扫描和掩码（3-d 数组），可通过以下方式访问：

import numpy as np

archive = np.load('Train set/Images/patient_002.npz')
train_scan = archive['scan']
train_mask = archive['mask']

test_set = np.load('Test set/Images/patient_000.npz')
test_scan = test_set['scan']
test_mask = test_set['mask']

将 test_scan.shape 和 train_mask.shape 作为 3d 矩阵：

输出：

(92, 92, 92)

【问题讨论】：

到底是什么问题？你在哪个部分苦苦挣扎？

标签： python load

【解决方案1】：

您可以使用os.walk。

import os
all_files = []
for _, _, file_name in os.walk('Train set/Images/'):
    all_files.extend(file_name)

all_files = [file_name for file_name in all_files if file_name.endswith('.npz')]

然后您可以在 for 循环中读取 all_files。希望这会有所帮助。

【讨论】：