читать несколько файлов с одинаковым именем, но с разным расширением Python - PullRequest
Новая
       10

читать несколько файлов с одинаковым именем, но с разным расширением Python

0 голосов
/ 27 августа 2018

Я пытаюсь читать и записывать файлы из папки с одинаковыми именами, НО разные расширения => извлекать из них данные => переписывать. Вот мой код: 

for header_name in glob.glob(os.path.join(folder_path, "*.json")):
    for nii_name in glob.glob(os.path.join(folder_path, "*.nii")):
        with open(header_name, "r") as f:
        nii_meta = json.load(f)
        add_stc = slicetime(header_name).tolist()
        nii_meta["SliceTiming"] = add_stc
    with open(header_name, 'w') as f:
        json.dump(nii_meta, f, indent=2)

я пытался сделать проверку: 

    h_name = os.path.splitext(os.path.basename(header_name))[0]
    n_name = os.path.splitext(os.path.basename(nii_name))[0]
    if h_name == n_name:
        do smth with data form files

но без успеха Имена файлов: sub01_T1w.json, sub01_T1w.nii.gz, sub01_T1w1.json, sub01_T1w1.nii.gz ...

1 Ответ

0 голосов
/ 27 августа 2018

Я бы предложил работать с pathlib , это относительно новая часть стандартной библиотеки python, которая немного облегчает работу с путями и файлами.Решение, использующее его, может выглядеть следующим образом: 

from pathlib import Path

p = Path('parent/folder/of/my/files')
json_names = {f.stem for f in p.iterdir() if f.suffix == '.json'}
nii_names = {Path(f.stem).stem for f in p.iterdir() if f.suffixes == ['.nii', '.gz']}
for file_name in json_names & nii_names:
    json_path = p / (file_name + '.json')
    nii_path = p / (file_name + '.nii.gz')
    with open(json_path) as json_file, open(nii_path) as nii_file:
        ...  # do things with the files' contents

Если вы хотите записать в них, не забудьте повторно открыть цель с помощью open(json_path, 'w') после выхода из with -блока только для чтения.

...