- 读文件和写文件
- 管道输出
- 读行和写行
- 使用FileInput对象读取文件
读文件和写文件
'''
读文件和写文件
r: 读
w: 写
r+：文件可读写，如果文件不存在，会抛出异常。如果文件存在，会从当前位置开始
写入新内容，通过seek函数可以改变当前的位置，也就是改变文件的指针。
w+：文件可读写，如果文件不存在，会创建一个新文件，如果文件存在，会清空整个
文件，并写入新内容。
a+：文件可读写，如果文件不存在，会创建一个新文件，如果文件存在，会将要写入的
内容添加到原文件的最后。也就是说，使用a+打开已经存在的文件，文件指针已经在文件的结尾了
'''
'''
write(string):向文件写入内容，该函数会返回写入文件的字节数。
read([n])：读取文件的内容，n是一个整数，表示从文件指针指定的位置开始读取的n个字节。
如果不指定n，该函数会读取从当前位置往后的所有的字节。该函数会返回读取的数据。
seek(n)：重新设置文件指针，也就是改变文件的当前位置。如果使用write函数写入内容后
需要使用seek(0)重置文件指针
close()：关闭文件，对文件进行读写操作后，关闭文件是一个好习惯。
'''
# 以写模式打开文件
# 目录不存在，会报错；文件不存在，会创建一个文件
f = open('./files/test1.txt', 'w')
# 向文件中写入内容， 返回的是写入的字节数
print(f.write('Life is short, ')) # 15
print(f.write('you need python.')) # 16
f.close()
# 读文件
f = open('./files/test1.txt', 'r')
print(f.read(15))
print(f.read(16))
f.close()
# r 文件不存在会抛出异常……废话吗
try:
f = open('./files/test2.txt', 'r')
except Exception as e:
print(e)
# a+ 模式
f = open('./files/test2.txt', 'a+')
f.write('hellokitty')
f.close()
# f = open('./files/text2.txt', 'a+') # 用a+模式打开文件，指针已经在最后了，所以第一个read没有结果
f = open('./files/test2.txt', 'r+') # r 和 r+ 模式打开，指针在最开始，两次都是是可以读取的
print('1 >>> ', f.read())
f.seek(0)
print('2 >>> ', f.read())
f.close()
# w+ 搭配 seek 使用
try:
f = open('./files/test2.txt', 'w+')
# f = open('./files/test2.txt', 'a+')
print(f.read())
f.write('Are you OK?')
f.seek(0)
print(f.read())
finally:
f.close()
管道输出
模拟了一个grep的操作：
'''
从标准输入读取所有的数据，并按行将数据保存到列表中，然后
过滤出所有包含“readme”的行，并输出这些行
'''
import sys, os, re
# 从标准输入读取全部数据
text = sys.stdin.read()
files = text.split(os.linesep)
for file in files:
result = re.match('.*readme.*', file)
if result != None: print(file)
读行与写行
import os
f = open('./files/urls.txt', 'r+')
url = ''
while True:
url = f.readline()
url = url.rstrip() # trim
if url == '': break
else: print(url)
print('============')
f.seek(0)
print(f.readlines())
f.write('www.baidu.com' + os.linesep)
f.close()
# 批量写入行
f = open('./files/urls.txt', 'a+')
urlList = ['www.zhihu.com' + os.linesep, 'www.weibo.com' + os.linesep]
#urlList = ['www.zhihu.com', 'www.weibo.com']
f.writelines(urlList)
f.close()
使用FileInput对象读取文件
import fileinput
fileobj = fileinput.input('./files/urls.txt')
print(type(fileobj)) # <class 'fileinput.FileInput'>
# print(fileobj.readline().rstrip())
# print(fileobj.readline().rstrip())
# print(fileobj.readline().rstrip())
for line in fileobj:
line = line.rstrip()
if line != '':
print(fileobj.lineno(), ' : ', line)
else:
print(fileobj.filename())
练习1
'''
编写一个Python程序，从控制台输入一个奇数，
然后生成奇数行的星号（*）菱形，并将该菱形保存到当前目录下
的stars.txt文件中。
'''
def fun(num, sep = '*'):
'''
输入一个奇数，打印菱形图案。
默认以'*'填充
'''
if num % 2 == 0: raise Exception('必须输入一个奇数')
# maxLine = int((num + 1)/2 - 1) # 宽度最大行的行号
# print('maxLIne: ', maxLine)
for line in range(num):
no = 0
if line < (num + 1)/2:
no = line
else:
no = num - line - 1
yield '{str:^{cnt}}'.format(str = (2 * no + 1) * sep, cnt = num)
import os
f = open('./starts.txt', 'w+')
for str in fun(9):
f.write(str + '\n')
f.close()
练习2
'''
编写一个Python程序，从当前目录的文本文件words.txt中读取所有
的内容（全都是英文单词），并统计其中每一个英文单词出现的次数。
单词之间用逗号（,）、分号（;）或空格分隔，也可能是这3个分隔符一起
分隔单词。将统计结果保存到字典中，并输出统计结果。
假设words.txt文件的内容如下：
test star test star star;bus test bill , new yeah bill,book bike God start python what
'''
import re
f = open('words.txt','r')
words = f.read()
wordList = re.split('[ ,;]+',words)
print(wordList)
countDict = {}
for word in wordList:
if countDict.get(word) == None:
countDict[word] = 1
else:
countDict[word] = int(countDict[word]) + 1
for (key,value) in countDict.items():
print(key,'=',value)
f.close()
