读文件
打开一个文件用open()方法(open()返回一个文件对象,它是可迭代的):
>>> f = open(“C:\\release\\testdata.txt”, ‘r’)
r表示是文本文件,rb是二进制文件。(这个mode参数默认值就是r)
如果文件不存在,open()函数就会抛出一个IOError的错误,并且给出错误码和详细的信息告诉你文件不存在:
>>> f = open(“C:\\release\\testdata.txt”, ‘r’)
Traceback (most recent call last):
File “”, line 1, in
FileNotFoundError: [Errno 2] No such file or directory: ‘C:\\release\\testdata.txt’
文件使用完毕后必须关闭,因为文件对象会占用操作系统的资源,并且操作系统同一时间能打开的文件数量也是有限的
>>> f.close()
try … finally 和 with
由于文件读写时都有可能产生IOError,一旦出错,后面的f.close()就不会调用。所以,为了保证无论是否出错都能正确地关闭文件,我们可以使用try … finally来实现:
try:
f = open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′)
print(f.read())
finally:
if f:
f.close()
但是每次都这么写实在太繁琐,所以,Python引入了with语句来自动帮我们调用close()方法:
with open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′) as f:
print(f.read())
三个“读”方法
python文件对象提供了三个“读”方法: read()、readline() 和 readlines()。每种方法可以接受一个变量以限制每次读取的数据量。
read() 每次读取整个文件,它通常用于将文件内容放到一个字符串变量中。如果文件大于可用内存,为了保险起见,可以反复调用read(size)方法,每次最多读取size个字节的内容。
readlines() 之间的差异是后者一次读取整个文件,象 .read() 一样。.readlines() 自动将文件内容分析成一个行的列表,该列表可以由 Python 的 for … in … 结构进行处理。
readline() 每次只读取一行,通常比readlines() 慢得多。仅当没有足够内存可以一次读取整个文件时,才应该使用 readline()。
注意:这三种方法是把每行末尾的’\n’也读进来了,它并不会默认的把’\n’去掉,需要我们手动去掉。
>> with open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′) as f1:
list1 = f1.readlines()
>> list1
[‘选择,填写,填写\n’, ‘选择,不填写,填写\n’, ‘不选择,填写,不填写\n’, ‘不选择,不填写,不填写’]
去掉’\n’
>>> with open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′) as f1:
list1 = f1.readlines()
for i in range(0, len(list1)):
list1[i] = list1[i].rstrip(‘\n’) #删除 string 字符串末尾的指定字符(默认为空格)
>>> list1
[‘选择,填写,填写’, ‘选择,不填写,填写’, ‘不选择,填写,不填写’, ‘不选择,不填写,不填写’]
对于read()和readline()也是把’\n’读入了,但是print的时候可以正常显示(因为print里的’\n’被认为是换行的意思)
>>> with open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′) as f1:
list1 = f1.read()
>>> list1
‘选择,填写,填写\n选择,不填写,填写\n不选择,填写,不填写\n不选择,不填写,不填写’
>>> print(list1)
选择,填写,填写
选择,不填写,填写
不选择,填写,不填写
不选择,不填写,不填写
>>> with open(“C:\\release\\testdata.txt”, ‘r’, encoding=’UTF-8′) as f1:
list1 = f1.readline()
>>> list1
‘111\n’
>>> print(list1)
111
高级一点的练习题
有两个文件,每个都有很多行ip地址,求出两个文件中相同的ip地址:
# coding:utf-8 排序模块
import bisect
with open(“C:\\release\\test1.txt”, ‘r’, encoding=’UTF-8′) as f1:
list1 = f1.readlines()
for i in range(0, len(list1)):
list1[i] = list1[i].strip(‘\n’)
with open(‘C:\\release\\test2.txt’, ‘r’, encoding=’UTF-8′) as f2:
list2 = f2.readlines()
for i in range(0, len(list2)):
list2[i] = list2[i].strip(‘\n’)
list2.sort()
length_2 = len(list2)
same_data = []
for i in list1:
pos = bisect.bisect_left(list2, i) # bisect_left函数是新元素会被放置于它相等的元素的前面,而 bisect_right返回的则是跟它相等的元素之后的位置。
if pos < len(list2) and list2[pos] == i:
same_data.append(i)
same_data = list(set(same_data)) # set() 函数创建一个无序不重复元素集
print(same_data)
要点就是:(1)用with (2)处理行末的’\n’ (3)使用二分查找提高算法效率。(4)使用set快速去重。
写文件
写文件和读文件是一样的,唯一区别是调用open()函数时,传入标识符’w’或者’wb’表示写文本文件或写二进制文件:
>>> f = open(‘C:\\release\\test.txt’, ‘w’, encoding=’UTF-8′) # 若是’wb’就表示写二进制文件
>>> f.write(‘Hello, world!’)
>>> f.close()
注意:’w’这个模式是这样的:
如果没有这个文件,就创建一个;
如果有,那么就会先把原文件的内容清空再写入新的东西。
所以若不想清空原来的内容而是直接在后面追加新的内容,就用’a’这个模式。
我们可以反复调用write()来写入文件,但是务必要调用f.close()来关闭文件。当我们写文件时,操作系统往往不会立刻把数据写入磁盘,而是放到内存缓存起来,空闲的时候再慢慢写入。只有调用close()方法时,操作系统才保证把没有写入的数据全部写入磁盘。忘记调用close()的后果是数据可能只写了一部分到磁盘,剩下的丢失了。所以,还是用with语句来得保险:
with open(‘test.txt’, ‘w’) as f:
f.write(‘Hello, world!’)
python文件对象提供了两个“写”方法: write() 和 writelines()。
write()方法和read()、readline()方法对应,是将字符串写入到文件中。
writelines()方法和readlines()方法对应,也是针对列表的操作。它接收一个字符串列表作为参数,将他们写入到文件中,换行符不会自动的加入,因此,需要显式的加入换行符。
f1 = open(‘test1.txt’, ‘w’)
f1.writelines([“1”, “2”, “3”])
# 此时test1.txt的内容为:123
f1 = open(‘test1.txt’, ‘w’)
f1.writelines([“1\n”, “2\n”, “3\n”])
# 此时test1.txt的内容为:
# 1
# 2
# 3