У меня есть несколько папок с данными, я хочу взять определенный файл 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...
Это также вернуло ошибку, указав имя файла и сказав, что оно не найдено. Я довольно смущен тем, что здесь происходит. Я мог бы использовать первый код и продолжать изменять имя пути, но это слишком много времени. Буду очень признателен за любые советы о том, где я ошибаюсь, или о том, как лучше подойти к проблеме.