Python脚本删除文本文件行中的特定单词

时间:2014-09-17 07:17:36

标签: python-2.7

我有一个带有一些站点链接的文本文件....我想删除站点名称之前的字符串。 这是输入文件>> input.txt中:

http://www.site1.com/
http://site2.com/
https://www.site3333.co.uk/
site44.com/
http://www.site5.com/
site66.com/

输出文件应该是:

site1.com/
site2.com/
site3333.co.uk/
site44.com/
site5.com/
site66.com/

这是我的代码:

bad_words = ['https://', 'http://', 'www.']

with open('input.txt') as oldfile, open('output.txt', 'w') as newfile:
    for line in oldfile:
        if not any(bad_word in line for bad_word in bad_words):
            newfile.write(line)
print './ done'

当我运行此代码时,它会完全删除包含bad_words的行

site44.com/
site66.com/

我应该如何处理代码才能获得具体结果?

1 个答案:

答案 0 :(得分:3)

谢谢所有我已经解决了这个......代码应该是:

fin = open("input.txt")
fout = open("output.txt", "w+")
delete_list = ['https://', 'http://', 'www.']
for line in fin:
    for word in delete_list:
        line = line.replace(word, "")
    fout.write(line)
fin.close()
fout.close()
print './ done'