天天學點Python Cookbook（三）

1.計算文件的行數

任務：
須要計算一個文件中有多少行。

解決方案：
用enumerate()方法再加循環計數進行統計，具體代碼以下：

def get_filelines(file_name):
    count = 0
    f = open(file_name,'r')
    for count,line in enumerate(f):
        count += 1
    f.close()
    return count

測試用例：

定義一個函數，向文件中寫入數字，每十個換一次行。

def write_file(file_name,line_number):
    with open(flie_name,'w') as f:
        for i in range(1,line_number):
            if i % 10 == 0:
                f.write('\n')
            f.write(str(i))

調用函數。

if __name__ == '__main__':
    flie_name = 'test.txt'
    line_number = 9999999
    write_file(flie_name,line_number)
    print(get_filelines(flie_name))

測試結果：