读写文件是最常见的IO操作。Python内置了读写文件的函数,用法和C是兼容的。
读文件
打开一个文件,读取后应该close(),为了保证无论是否出错都能正确地关闭文件,我们可以使用try ... finally来实现:
1. try:
2. '/path/to/file', 'r')
3. print f.read()
4. finally:
5. if f:
6. f.close()
Python引入了with语句来自动帮我们调用close()方法:
1. <pre name="code" class="python">with open('/path/to/file', 'r') as f:
2. print f.read()
这和前面的try ... finally是一样的,但是代码更佳简洁,并且不必调用f.close()方法。调用read()方法可以一次读取文件的全部内容,Python把内容读到内存,用一个
str对象
表示
读大文件时,建议下面的方式:
1. with open('/path/to/file', 'r') as f:
2. for line in f:
3. print line
这种迭代的方式不会一次性的把大文件都读入内存。
二进制文件
前面讲的默认都是读取文本文件,并且是ASCII编码的文本文件。要读取二进制文件,比如图片、视频等等,用'rb'模式打开文件即可。
1. >>> f = open('/Users/michael/test.jpg', 'rb')
2. >>> f.read()
3. '\xff\xd8\xff\xe1\x00\x18Exif\x00\x00...' # 十六进制表示的字节
字符编码
要读取非ASCII编码的文本文件,就必须以二进制模式打开,再解码。比如GBK编码的文件:
1. >>> f = open('/Users/michael/gbk.txt', 'rb')
2. >>> u = f.read().decode('gbk')
3. >>> u
4. u'\u6d4b\u8bd5'
5. >>> print u
6. 测试
如果每次都这么手动转换编码嫌麻烦(写程序怕麻烦是好事,不怕麻烦就会写出又长又难懂又没法维护的代码),Python还提供了一个
codecs模块
帮我们在读文件时自动转换编码,
直接读出unicode
:
1. import codecs
2. with codecs.open('/Users/michael/gbk.txt', 'r', 'gbk') as f:
3. # u'\u6d4b\u8bd5'
以gbk编码打开文件/Users/michael/gbk.txt,
读取后直接自动转换成python内部的unicode编码
。
写文件
写文件和读文件是一样的,唯一区别是调用open()函数时,传入标识符'w'或者'wb'表示写文本文件或写二进制文件:
1. >>> f = open('/Users/michael/test.txt', 'w')
2. >>> f.write('Hello, world!')
3. >>> f.close()
你可以反复调用write()来写入文件,但是务必要调用f.close()来关闭文件。当我们写文件时,操作系统往往不会立刻把数据写入磁盘,而是放到内存缓存起来,空闲的时候再慢慢写入。只有调用close()方法时,操作系统才保证把没有写入的数据全部写入磁盘。忘记调用close()的后果是数据可能只写了一部分到磁盘,剩下的丢失了。所以,还是用with语句来得保险:
1. with open('/Users/michael/test.txt', 'w') as f:
2. 'Hello, world!')
要写入特定编码的文本文件,请效仿codecs的示例,在open()指定以什么编码打开文件,在写入时将待写入内容先解码成unicode,既(str.decode("unicode_escape")),那么在write()时,codecs就会自动将内容转换成指定编码写入文件。