在 Python 中，我們平時(shí)處理的輸入輸出（I/O）大多數(shù)是針對(duì)文件或網(wǎng)絡(luò)數(shù)據(jù)的，但有時(shí)候，我們希望直接在內(nèi)存中進(jìn)行讀寫，而不必先把數(shù)據(jù)存到硬盤上。
這時(shí)候，StringIO 和 BytesIO 就派上用場(chǎng)了。

簡(jiǎn)單理解：

• StringIO → 操作內(nèi)存中的 字符串（str 類型）
• BytesIO → 操作內(nèi)存中的 二進(jìn)制數(shù)據(jù)（bytes 類型）

它們都來自 io 模塊，接口和文件對(duì)象幾乎一模一樣，所以學(xué)會(huì)文件操作就能直接用它們。

1. 為什么要用 StringIO / BytesIO

• 無需磁盤 I/O，速度快
• 便于測(cè)試（不需要真實(shí)文件）
• 在網(wǎng)絡(luò)編程中常用（比如 HTTP 響應(yīng)先存在內(nèi)存中再處理）
• 數(shù)據(jù)臨時(shí)存儲(chǔ)（比如生成報(bào)告、圖片）

2. StringIO 基礎(chǔ)用法

StringIO 適合處理 文本數(shù)據(jù)，使用方式類似普通文件：

from io import StringIO

# 創(chuàng)建 StringIO 對(duì)象
f = StringIO()

# 寫入數(shù)據(jù)
f.write("Hello ")
f.write("Python")

# 獲取當(dāng)前內(nèi)容
print(f.getvalue())  # Hello Python

注意：getvalue() 用來獲取緩沖區(qū)的全部數(shù)據(jù)。

3. StringIO 讀取數(shù)據(jù)

我們也可以直接用 StringIO 來模擬讀取文件：

from io import StringIO

data = "Line1\nLine2\nLine3"
f = StringIO(data)

# 一次性讀取
print(f.read())

# 重置游標(biāo)
f.seek(0)

# 按行讀取
for line in f:
    print(line.strip())

要點(diǎn)：

• seek(0) → 把“文件指針”移動(dòng)到開頭
• 讀取方法（read、readline、readlines）和文件對(duì)象一致

4. BytesIO 基礎(chǔ)用法

BytesIO 用于處理 二進(jìn)制數(shù)據(jù)，比如圖片、音頻、壓縮包等。

from io import BytesIO

# 創(chuàng)建 BytesIO
f = BytesIO()

# 寫入二進(jìn)制數(shù)據(jù)
f.write(b"Hello World")

# 獲取數(shù)據(jù)
print(f.getvalue())  # b'Hello World'

5. BytesIO 讀取數(shù)據(jù)

from io import BytesIO

data = b"Binary\x00Data"
f = BytesIO(data)

# 讀取前6個(gè)字節(jié)
print(f.read(6))  # b'Binary'

# 繼續(xù)讀剩下的
print(f.read())   # b'\x00Data'

6. StringIO vs BytesIO 區(qū)別

7. 從文件到內(nèi)存的轉(zhuǎn)換

有時(shí)我們需要把磁盤文件內(nèi)容讀到內(nèi)存中用 BytesIO 處理，比如處理圖片：

from io import BytesIO

# 假設(shè)有個(gè)圖片
with open("test.png", "rb") as f:
    data = f.read()

# 存入 BytesIO
bio = BytesIO(data)

# 讀取前10個(gè)字節(jié)
print(bio.read(10))

8. 從內(nèi)存保存到文件

反過來，我們也可以把 BytesIO 或 StringIO 的數(shù)據(jù)寫到磁盤：

from io import BytesIO

# 創(chuàng)建內(nèi)存數(shù)據(jù)
bio = BytesIO()
bio.write(b"Save me to file")

# 寫入文件
with open("output.bin", "wb") as f:
    f.write(bio.getvalue())

9. 常見使用場(chǎng)景

1. 臨時(shí)構(gòu)建數(shù)據(jù)，避免磁盤 I/O
2. 測(cè)試代碼時(shí)，模擬文件對(duì)象
3. 網(wǎng)絡(luò)數(shù)據(jù)處理（HTTP 請(qǐng)求、Socket 傳輸）
4. 圖片或音頻處理（配合 Pillow、pydub 等庫(kù)）

10. 最佳實(shí)踐

• 如果處理的是文本，用 StringIO，并且直接寫 str 類型。
• 如果處理的是二進(jìn)制數(shù)據(jù)（比如圖片、壓縮包），用 BytesIO，并且確保寫入的是 bytes。
• 操作完成后不一定非要 close()（因?yàn)樗鼈兪莾?nèi)存對(duì)象），但為了習(xí)慣最好加上。
• 需要頻繁讀取時(shí)，記得用 seek(0) 重置指針。

到此這篇關(guān)于Python中StringIO和BytesIO用法及區(qū)別的文章就介紹到這了,更多相關(guān)Python StringIO BytesIO內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！ 

最新評(píng)論