Python IO操作(10)

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')
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。
禁止轉(zhuǎn)載,如需轉(zhuǎn)載請(qǐng)通過(guò)簡(jiǎn)信或評(píng)論聯(lián)系作者。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容