Блог

Я написал фрагмент кода, который выполняет поиск каталогов для текстового файла и на основе условия if, которое я написал, копирует, сохраняет и записывает в новый файл. Проблема, с которой я борюсь, заключается в том, что когда разные условия if добавляются к одному и тому же пустому списку, он сохраняет выходные данные второго «условия if» в виде нового массива элементов. Однако я хочу, чтобы выходные данные нескольких условий if сохранялись в одной строке. Чтобы быть более конкретным, вот как мой код сохраняет и записывает в текстовый файл.

 import matplotlib.pyplot as plt
import numpy as np
import math
from scipy.optimize import curve_fit
from IPython import embed

import os

File = 'results.txt'
ospath = os.walk('/Users/analyses/')

mylist=[]

for subdir, dirs, files in ospath:
    if File in files:
        filepath = subdir   '/'   File
        with open(filepath) as fp:
            for line in fp:
                line_list = []
                if "Bckgnd subtracted source c/r" in line:
                    line_array = line.split(" ")
                    mylist.extend((line_array[5],line_array[7].rstrip("n")))
                elif "Bckgnd subtracted source cnts" in line:
                    line_array = line.split(" ")
                    mylist.extend((line_array[5],line_array[7].rstrip("n")))
                mylist.append(line_list)

mylist = np.asarray(mylist)


with open("mylist.txt", "w") as txt_file:
    for line in mylist:
        txt_file.write('n'.join(','.join(elems) for elems in mylist))
  

Вывод :

     11.3,4.8
    id_06,BANANA
    9.3,5.6
    id_01,APPLE
    8.7, 23.8
    id_08, ORANGE
    
    mylist[0] = [11.3,4.8] #comes from first if condition
    mylist[1] = [id_06,BANANA] #comes from second if condition
  

Где так, как я хочу:

     11.3,4.8,id_06,BANANA 
    9.3,5.6,id_01,APPLE
    8.7, 23.8, id_08, ORANGE
  

где mylist[0] = [11.3,4.8,id_06,BANANA]

Не совсем уверен, подходит ли это для вашей проблемы. Но вы можете попробовать следующее:

 mylist = []
with open(filepath) as fp:
    for line in fp:
        line_list = []
        if "condition1" in line:
            line_array = line.split(" ")
            line_list.extend([line_array[5],line_array[7].rstrip("n")])
        if "condition2" in line:
            line_array = line.split(" ")
            line_list.extend((line_array[5],line_array[7].rstrip("n")))
        mylist.append(line_list)

mylist = np.asarray(mylist)
  

При этом используется отдельный список для каждой строки, который расширяется, чтобы получить единый список, а не список списков, как это произойдет при добавлении данных. Список очищается для каждой строки.
Возможно, это не самый быстрый или элегантный способ сделать это, но он должен решить вашу проблему.

Похоже, что несколько отступов в вашем примере кода могут быть неправильными — кажется, что for line in fp: строка должна иметь отступ на один уровень меньше, а mylist.append(...) строки должны иметь отступ на один уровень больше.

Что касается вашего вопроса: если я правильно понимаю, вам нужен line_array из обоих условий if в одной строке вашего mylist.txt файл. Предполагая, что это так, попробуйте использовать array.extend(...) :

 with open(filepath) as fp:
    for line in fp:
        line_array = []
        if "condition1" in line:
            line_array = line.split(" ")
            mylist.extend([line_array[5],line_array[7].rstrip("n")])
        if "condition2" in line:
            line_array = line.split(" ")
            mylist.append([line_array[5],line_array[7].rstrip("n")])