文章来自《Python cookbook》.

翻译仅仅是为了个人学习,其它商业版权纠纷与此无关!

-- 61.182.251.99 [2004-09-20 19:38:11]

1. 描述

文件中查找替换字符串

1.1. 问题

替换文件中特定字符串

1.2. 解决

使用string的replace方法可以简单实现字符串替换操作。从指定文件(或标准输入)读入内容,进行处理,然后写入指定文件(或标准输出),工作就完成了。

   1 import os, sys
   2 
   3 nargs = len(sys.argv)
   4 
   5 if not 3 <= nargs <= 5:
   6     print "usage: %s search_text replace_text [infile [outfile]]" % \
        os.path.basename(sys.argv[0])
   7 else:
   8     stext = sys.argv[1]
   9     rtext = sys.argv[2]
  10     input = sys.stdin                                     #标准输入  
  11     output = sys.stdout                                   #标准输出 
  12     if nargs > 3:
  13         input = open(sys.argv[3])                         #数据源文件 
  14     if nargs > 4:
  15         output = open(sys.argv[4], 'w')                   #数据目的文件  
  16 
  17     for s in input.xreadlines(  ):                        #读入文件的每一行
  18         output.write(s.replace(stext, rtext))             #处理,并写入目的文件 
  19     output.close(  )
  20     input.close(  )

1.3. 讨论 Discussion

处理方法很简单,这也是赞的地方。杀鸡不用牛刀。上面的"漂亮"代码是可以执行的脚本。 脚本通过查询参数,确定被替换的字符串,替换字符串,以及输入文件(默认是标准输入), 输出文件(默认是标准输出)。然后循环处理从输入文件读出的每一行,替换字符串,写入输出文件。就这么简单!代码很漂亮,是吧? 还有,脚本最后关闭了两个文件。

如果有足够的内存可以缓存原来的字符串和替换处理后的字符串(字符串是不变对象), 那么有办法进行高效处理。 可以不用循环, 而是一次将原文件内容读入到临时字符串, 在等同于整个文件内容的这个字符串上执行replace操作, 然后写入输出文件。 现在典型的PC机有256M内存,处理100M大小的文件应该没有问题, 用下面的语句替代循环所需要的内存足够了

   1 output.write(input.read(  ).replace(stext, rtext))

这一条语句较之上面的循环处理又简单一些。

如果还在使用Python的早期版本,比如1.5.2 ,方法如下:

修改import 语句为:

   1         import os, sys, string

修改最后两条语句为:

   1 for s in input.readlines(  ):
   2     output.write(string.replace(s, stext, rtext))

文件的xreadlines方法是Python 2.1中引入的。它采取措施避免一次将整个文件内容全部读入内存。readlines 方法必须将文件内容一次读入内存,在处理大文件时可能有麻烦。

Python 2.2中, 更直观的代码如下:

   1 for s in input:
   2     output.write(s.replace(stext, rtext))

这是最快最简单的方法.

#译注 : file.xreadlines() is deprecated since Python 2.3, use the following instead

   1            for line in file:
   2                process(line)

#end 译注

1.4. 参考 See Also

Python 文档内置函数openfile 对象部分