IO操作,即對(duì)硬盤上的數(shù)據(jù)進(jìn)行讀寫操作。
直接上代碼:
try:
# 'r'表示讀取文件,encoding表示編碼,errors='ignore'可以忽略錯(cuò)誤編碼,避免報(bào)錯(cuò)UnicodeDecodeError
f = open('F:\\233.txt', 'r', encoding='utf-8', errors='ignore')
# 一次性讀取所有內(nèi)容
# print(f.read())
# 一次最多讀取n個(gè)字節(jié),這里n=1024
# print(f.read(1024))
# 一次讀取一行
# print(f.readline())
# 一次讀取所有行,返回list
for line in f.readlines():
# strip()去除末尾的換行符\n
print(line.strip())
finally:
if f:
# 關(guān)閉
f.close()
對(duì)于讀寫操作,最后一定要關(guān)閉文件對(duì)象,因?yàn)槲募?duì)象會(huì)占用操作系統(tǒng)的資源。代碼不夠簡(jiǎn)潔,Python支持with語(yǔ)句自動(dòng)調(diào)用close()方法,代碼如下:
with open('F:\\233.txt', 'r') as f:
f.read()
如果要讀取二進(jìn)制文件,比如圖片、視頻等,要用'rb'模式。
with open('F:\\原型圖\\Book detail.png', 'rb') as b:
# 輸出十六進(jìn)制表示的字節(jié)
b.read()
寫文件跟讀文件類似,'w'表示寫文本文件, 'wb'表示寫二進(jìn)制文件
# 如果abc.txt已經(jīng)存在,那么會(huì)先刪除,再重建一個(gè)abc.txt
with open('F:\\abc.txt', 'w', encoding='utf-8') as wf:
wf.write('Hello')
如果要在文件末尾追加內(nèi)容呢?這時(shí)候需要'a'模式,代碼如下:
with open('F:\\233.txt', 'a', encoding='utf-8') as af:
af.write('Hello, 233')
StringIO類
StringIO類有兩種方式往內(nèi)存中寫str。
第一種:
from io import StringIO
# StringIO 在內(nèi)存中讀寫str
string_io = StringIO()
string_io.write('Hello')
string_io.write(', ')
string_io.write('World')
# 獲取寫入的str
print('StringIO: ', string_io.getvalue())
第二種:
s = StringIO('Hello!\n Read! ')
while True:
# 可以像讀文件一樣讀取內(nèi)容
line = s.readline()
if not line:
break
print(line.strip())
BytesIO類
跟StringIO類似,只不過(guò)StringIO是寫入str,BytesIO是寫入byte。
這里需要注意getvalue()方法和read()方法因?yàn)閟tream position而產(chǎn)生的差異,可以通過(guò)tell()獲取position,也可以通過(guò)seek()調(diào)整position。
from io import BytesIO
# 在內(nèi)存中讀寫二進(jìn)制數(shù)據(jù)
bytes_io = BytesIO()
bytes_io.write('天朝'.encode('utf-8'))
print('position已經(jīng)移到%d' % bytes_io.tell())
print(bytes_io.getvalue().decode('utf-8'))
# write后,stream position已經(jīng)移到6,所以接下來(lái)read讀取不到內(nèi)容
print('seek之前,', bytes_io.read().decode('utf-8'))
# seek調(diào)整stream position
bytes_io.seek(0)
print('seek之后,', bytes_io.read().decode('utf-8'))
b = BytesIO('天朝'.encode('utf-8'))
# 此時(shí)stream position為0
print('position已經(jīng)移到%d' % b.tell())
print(b.read().decode('utf-8'))
OS類
import os
# nt代表windows,posix代表Linux、UNIX或Mac OS
print(os.name)
當(dāng)前目錄的絕對(duì)路徑
print(os.path.abspath('.'))
當(dāng)前目錄的相對(duì)路徑
print(os.path.relpath('F:/pythonCode/pythonCode.iml', 'F:'))
如果要在某個(gè)目錄下創(chuàng)建一個(gè)新目錄,首先獲取新目錄的完整路徑(此時(shí)并沒(méi)有創(chuàng)建出目錄)
path_join = os.path.join('F:/Z', 'ZZZZ')
判斷目錄是否存在
if os.path.exists(path_join):
# 刪除目錄
os.rmdir(path_join)
創(chuàng)建目錄
os.mkdir(path_join)
os.path.split把路徑拆分為兩部分,后一部分一定是最后級(jí)別的目錄或者文件名
print(os.path.split(path_join))
print(os.path.split('F:/233.txt'))
輸出:
('F:/Z', 'ZZZZ')
('F:/', '233.txt')
os.path.splitext把路徑拆分為兩部分,后一部分一定是文件后綴名或者''
print(os.path.splitext(path_join))
print(os.path.splitext('F:/666.txt'))
輸出:
('F:/Z\\ZZZZ', '')
('F:/666', '.txt')
上面合并、拆分的函數(shù)并不要求目錄和文件一定存在,它們只對(duì)字符串操作。
重命名文件
os.rename('F:/555.txt', 'F:/888.txt')
刪除文件
os.remove('F:/888.txt')
復(fù)制移動(dòng)文件等功能在shutil模塊中
import shutil
# 復(fù)制文件
shutil.copy('F:/233.txt', 'F:/Z/233.txt')
# 移動(dòng)文件
# shutil.move('F:/666.txt', 'F:/Z/666.txt')
獲取當(dāng)前目錄下所有目錄名,不包含目錄的路徑
print([x for x in os.listdir('.') if os.path.isdir(x)])
獲取當(dāng)前目錄下所有后綴名為.py的文件名,不包含文件的路徑
print([x for x in os.listdir('.') if os.path.isfile(x) and os.path.splitext(x)[1] == '.py'])
獲取某個(gè)目錄下文件名包含'Test'字符串的文件相對(duì)路徑:
def get_file(path='.'):
if os.path.exists(path):
if os.path.isdir(path):
dirs = os.listdir(path)
for v in dirs:
# v只是當(dāng)前目錄下的目錄名或者文件名,需重新組合路徑
join = os.path.join(path, v)
if os.path.isfile(v):
if v.find('Test') != -1:
print(join)
else:
get_file(join)
else:
if path.find('Test') != -1:
print(path)
else:
print('path is not exists.')
# 調(diào)用函數(shù),傳入目錄
get_file('F:\\pythonCode')