Python: Как извлечь данные в текстовом файле на основе информации о классе из другого текстового файла?

Question

В этом случае есть 3 класса, которые представлены значениями 0, 1 и 2. Я хотел бы извлечь информацию, относящуюся к классу 1 из другого текстового файла с именем fileA.txt. Я хотел бы знать, как это решить, используя python.

Например:

class.txt

0 
0 
1 
2 
2 
1 
1 

fileA.txt

a=[1,3,2,1] 
b=[3,2] 
c=[3,2,1] 
d=[3,3] 
e=[4,5,6] 
f=[3,2,3] 
g=[2,2] 

Ожидаемый результат:

c=[3,2,1] 
f=[3,2,3] 
g=[2,2] 

Может ли кто-Хель p me?

Answer 1

Прочитайте файл "class.txt" и создать список классов:

with open("class.txt", "rt") as f: 
    classes = [int(line) for line in f.readlines()] 

Прочитайте файл "fileA.txt" и создать список правильных линий:

with open("fileA.txt", "rt") as f: 
    lines = [line for index, line in enumerate(f.readlines()) if classes[index] == 1] 

Показать результаты:

print "".join(lines) 

Answer 2

не решение Python, но мне это нравится :)

$ grep -n "^1$" class.txt | cut -d: -f1 | while read linenumber 
do 
    sed -n "${linenumber}p" < fileA.txt 
done 

Выходные:

c=[3,2,1] 
f=[3,2,3] 
g=[2,2] 

Инструменты, используемые являются:

• grep
• cut
• sed

Answer 3

Это intu itive способ сделать это

classes = [l.strip() for l in open("class.txt").readlines()] 
indices = [i for i, x in enumerate(classes) if x == "1"] 

with open('fileA.txt') as file: 
    for index,line in enumerate(file): 
     if(index in indices): 
      print(line)