從檔案中讀取資料

讀取整個檔案

這裡假設在當前目錄下有一個檔名為’pi_digits.txt’的文字檔案，裡面的資料如下：

3.1415926535
8979323846
2643383279

with open('pi_digits.txt') as f: # 預設模式為‘r’，只讀模式
    contents = f.read() # 讀取檔案全部內容
    print contents # 輸出時在最後會多出一行（read()函式到達檔案末會返回一個空字元，顯示出空字元就是一個空行）
    print '------------'
    print  contents.rstrip() # rstrip()函式用於刪除字串末的空白
 

3.1415926535
8979323846
2643383279

------------
3.1415926535
8979323846
2643383279

逐行讀取

可以通過迴圈來實現逐行讀取資料：

with open('pi_digits.txt') as f:
    for line1 in f:
        print line1 # 每行末尾會有一個換行符
    print '------------'
    for line2 in f:
        print line2.rstrip() # 此時檔案已經讀完，line2指向文字末尾，因此不會有輸出
 

3.1415926535

8979323846

2643383279

------------

讀取檔案時相當於有一個指標在記錄讀取的位置，資料讀到哪，這個指標就指到哪邊，繼續讀取資料時會從該位置繼續讀取，因此上面程式碼中第二個迴圈中輸出為空。將上述程式碼稍加修改如下：

with open('pi_digits.txt') as f:
    for line1 in f:
        print line1
    print '------------'

with open('pi_digits.txt') as f: # 需要重新開啟文字進行讀取
    for line2 in
 
 f:
        print line2.rstrip() # 刪除字串末尾的空白

3.1415926535

8979323846

2643383279

------------
3.1415926535
8979323846
2643383279

上述程式碼相當於第一次讀取完後關閉該檔案又重新開啟進行讀取。逐行讀取資料也可以用readline()函式，如下：

with open('pi_digits.txt') as f: 
    # readline()每一次讀取一行資料，並指向該行末尾
    line1 = f.readline() # 讀取第一行資料（此時已經指向第一行末尾）
    line2 = f.readline() # 從上一次讀取末尾開始讀取（第二行）

    print line1.rstrip()
    print line2.rstrip()

3.1415926535
8979323846

有時候我們想要一次性讀取全部資料並且按分開儲存以便於後續的操作，當然用上面的迴圈可以實現，但python提供了更簡單的方法readlines()：

with open('pi_digits.txt') as f: 
    lines = f.readlines() # 讀取文字中所有內容，並儲存在一個列表中，列表中每一個元素對應一行資料
print lines # 每一行資料都包含了換行符

print '------------'
for line in lines:
    print line.rstrip()   
print '------------'
pi_str = '' # 初始化為空字元
for line in lines:
    pi_str += line.rstrip() #字串連線
print pi_str

['3.1415926535\n', '8979323846\n', '2643383279\n']
------------
3.1415926535
8979323846
2643383279
------------
3.141592653589793238462643383279

寫資料到檔案

寫資料有幾種不同的模式，最常用的是w’, ‘a’, 分別表示擦除原有資料再寫入和將資料寫到原資料之後：

filename = 'write_data.txt'
with open(filename,'w') as f: # 如果filename不存在會自動建立， 'w'表示寫資料，寫之前會清空檔案中的原有資料！
    f.write("I am Meringue.\n")
    f.write("I am now studying in NJTECH.\n")

此時會在當前路徑下建立一個’write_data.txt’的文字檔案，並向檔案中寫入資料如下：

I am Meringue.
I am now studying in NJTECH.

下面繼續在該檔案中加入新資料：

with open(filename,'a') as f: # 'a'表示append,即在原來檔案內容後繼續寫資料（不清楚原有資料）
    f.write("I major in Machine learning and Computer vision.\n")

此時的檔案內容為：

I am Meringue.
I am now studying in NJTECH.
I major in Machine learning and Computer vision.