L oop для построения нескольких текстовых файлов из разных папок не могу найти .txt файлы - PullRequest
0 голосов
/ 04 мая 2020

У меня есть несколько папок с данными, я хочу взять определенный файл c из каждой папки и построить его. У меня есть код, чтобы сделать это для одного файла:

data = np.genfromtxt('path to data')
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111)
im = ax.imshow(data, aspect = 'auto', cmap = 'plasma', interpolation = 'none', origin = 'lowest', vmax=10)
etc...

И у меня также есть код для перебора папок и получения необходимых мне файлов и добавления их в массив:

for sub in subfolders:
    for files in os.listdir(sub):
        if (files.startswith("data") and files.endswith(".txt")):
            file_list.append(files)
print(file_list)        

for i in file_list:
    data = np.genfromtxt(i)
    fig = plt.figure(figsize=(8, 6))
    ax = fig.add_subplot(111)
etc...

Когда я печатал массив, он имел все имена файлов, но во втором блоке, где я пытался построить его, он сказал, что файл не может быть найден, поэтому я попытался сделать это вместо этого:

for sub in subfolders:
    for files in os.listdir(sub):
        if (files.startswith("data") and files.endswith(".txt")):
            data = np.genfromtxt(files)
            fig = plt.figure(figsize=(6, 4))
            ax = fig.add_subplot(111)
            etc...

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

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...