Python: 如何根据来自另一个文本文件的class信息提取文本文件中的数据？

Question

在这种情况下，有 3 个 classes，由值 0、1 和 2 表示。我想从另一个名为的文本文件中提取属于 class 1 的信息fileA.txt。我想知道如何使用 python 解决这个问题。

例如：

class.txt

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]

谁能帮帮我？

Answer 1

不是 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]

使用的工具是：

Answer 2

读取 "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 3

这是一种直观的方法

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)

Python: 如何根据来自另一个文本文件的class信息提取文本文件中的数据？

Python: How to extract data in text file based on class information from another text file?

python

extract

extraction

data-extraction