Python使用来自另一个文件的输入搜索文件中的文本

Question

我是 python 和编程的新手。我需要一些关于 python 脚本的帮助。有两个文件，每个文件都包含电子邮件地址（超过 5000 行）。输入文件包含我想在数据文件中搜索的电子邮件地址（也包含电子邮件地址）。然后我想将输出打印到文件或显示在控制台上。我搜索脚本并能够修改，但我没有得到想要的结果。你能帮帮我吗？

dfile1 (50K lines)
yyy@aaa.com
xxx@aaa.com
zzz@aaa.com


ifile1 (10K lines)
ccc@aaa.com
vvv@aaa.com
xxx@aaa.com
zzz@aaa.com

Output file
xxx@aaa.com
zzz@aaa.com



datafile = 'C:\\Python27\\scripts\\dfile1.txt'
inputfile = 'C:\\Python27\\scripts\\ifile1.txt'

with open(inputfile, 'r') as f:
names = f.readlines()

outputlist = []

with open(datafile, 'r') as fd:
  for line in fd:
    name = fd.readline()
    if name[1:-1] in names:
        outputlist.append(line)
    else:
        print "Nothing found"
 print outputlist

新代码

with open(inputfile, 'r') as f:
    names = f.readlines()
outputlist = []

with open(datafile, 'r') as f:
    for line in f:
        name = f.readlines()
        if name in names:
            outputlist.append(line)
        else:
            print "Nothing found"
    print outputlist

Answer 1

也许我遗漏了一些东西，但为什么不使用一对呢？

#!/usr/local/cpython-3.3/bin/python

data_filename = 'dfile1.txt'
input_filename = 'ifile1.txt'

with open(input_filename, 'r') as input_file:
    input_addresses = set(email_address.rstrip() for email_address in input_file.readlines())

with open(data_filename, 'r') as data_file:
    data_addresses = set(email_address.rstrip() for email_address in data_file.readlines())

print(input_addresses.intersection(data_addresses))

Answer 2

mitan8 给出了你遇到的问题，但我会这样做：

with open(inputfile, "r") as f:
    names = set(i.strip() for i in f)

output = []

with open(datafile, "r") as f:
    for name in f:
        if name.strip() in names:
            print name

这样可以避免将较大的数据文件读入内存。

如果您想写入输出文件，您可以在第二个with 语句中执行此操作：

with open(datafile, "r") as i, open(outputfile, "w") as o:
    for name in i:
        if name.strip() in names:
            o.write(name)

Answer 3

我会这样做：

names=[]
outputList=[]
with open(inputfile) as f:
    for line in f:
        names.append(line.rstrip("\n")

myEmails=set(names)

with open(outputfile) as fd, open("emails.txt", "w") as output:
    for line in fd:
        for name in names:
            c=line.rstrip("\n")
            if name in myEmails:
                print name #for console
                output.write(name) #for writing to file

Answer 4

我认为您的问题源于以下几点：

name = fd.readline()
if name[1:-1] in names:

name[1:-1] 对每个电子邮件地址进行切片，以便您跳过第一个和最后一个字符。虽然在“dfile”中加载名称数据库时，通常跳过最后一个字符（换行符'\n'）可能会很好

with open(inputfile, 'r') as f:
    names = f.readlines()

您正在包括换行符。所以，根本不要对“ifile”中的名称进行切片，即

if name in names:

Answer 5

我认为您可以删除 name = fd.readline()，因为您已经在 for 循环中找到了该行。除了每次读取一行的 for 循环之外，它还会读取 another 行。另外，我认为name[1:-1] 应该是name，因为您不想在搜索时去掉第一个和最后一个字符。 with 自动关闭打开的文件。

PS：我会怎么做：

with open("dfile1") as dfile, open("ifile") as ifile:
    lines = "\n".join(set(dfile.read().splitlines()) & set(ifile.read().splitlines())
print(lines)
with open("ofile", "w") as ofile:
    ofile.write(lines)

在上述解决方案中，基本上我将两个文件的行的并集（两个集合的元素部分）来查找公共行。