StringIO和BytesIO的用法

數據讀寫有兩種方式：

1、直接讀寫。案例

import openpyxl

def write_excel():
    f = openpyxl.Workbook()  # 創建工作簿
    # sheet1 = f.create_sheet()
    print("sheet names:", f.sheetnames)
    sheet1 = f.active   # 當前工作表的名稱
    print("sheet1:", sheet1)
    row_file = 5 # 生成5行
    col_three = 3 # 生成3列

    for row in range(row_file):
        for col in range(col_three):
            rw = row + 1
            cl = col + 1
            if col % 2 == 0:
                sheet1.cell(row=rw, column=cl, value='1')
            else:
                sheet1.cell(row=rw, column=cl, value='2')

    f.save("OPENPYXL.xlsx")  # 保存文件

if __name__ == '__main__':
    # 寫入Excel
    write_excel()
print('openpyxl 寫入成功')

 

2、在內存中讀寫。[StringIO讀寫是字符串，  BytesIO讀寫是二進制文件]

import StringIO

output = StringIO.StringIO()
output.write('First line.\n')
print >>output, 'Second line.'

# Retrieve file contents -- this will be
# 'First line.\nSecond line.\n'
contents = output.getvalue() #獲取寫入的數據

# Close object and discard memory buffer --
# .getvalue() will now raise an exception.
output.close()  #最后關閉

from io import StringIO
ff = StringIO('Hello!\nHi!\nGoodbye!')
f = StringIO()  #創建StringIO對象--可以看做在內存中的文件
i=f.write('hello我是中國人\n') #寫入字符串，自動移動讀寫指針
#返回值：返回字符串個數
i=f.write('李明\n')
f.write('中國\n')
str=f.getvalue() #讀取字符串(全部)
#不操作指針
 
ss=f.seek(0) #移動讀寫指針
"""參數2：
為0時代表從文件開頭開始算起
1代表從當前位置開始算起
2代表從文件末尾算起 ??
如果操作成功，則返回新的指針位置，如果操作失敗，則函數返回 -1
"""
s=f.tell() #返回當前讀寫位置
# s = f.readline() #從當前位置開始讀取一行，包括\n；自動移動讀寫指針
#s = f.readline(3) #從當前位置開始讀取一行，包括\n；自動移動讀寫指針
# 參數 讀取的最大字符數
 
#s=f.readlines()  #讀取所有行，返回一個字符串列表;；自動移動讀寫指針
#['hello我是中國人\n', '李明\n', '中國\n']
 
#s = f.readlines(2) #讀取指定行，返回一個字符串列表;自動移動讀寫指針
 
#f.flush() #刷新
#s=f.read() # 讀取全部內容;自動移動讀寫指針
#s=f.read(5)  #從當前位置開始讀取的最大字符數
#f.truncate(3) #保留最前面的n個字符數，其它都刪除
 
f.close() #關閉

from io import BytesIO
 
# 方式一
f = BytesIO()  #創建BytesIO對象
i=f.write('中文\n'.encode('utf-8')) #寫入字符串--字節形式
#返回值：寫入的字節數；注意不是字符數
 
f.write('中國'.encode('utf-8'))
 
x=f.getvalue() #獲取字符串(全部)--字節形式
# b'\xe4\xb8\xad\xe6\x96\x87'
 
 
# 方式二：可以用一個bytes初始化BytesIO，然后，像讀文件一樣讀取
ff = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87')
y=ff.read() #讀取字節
 
print(y)

優點：StringIO和BytesIO是在內存中操作str和bytes的方法，使得和讀寫文件具有一致的接口。