每天学点Python Cookbook（三）

2019年8月15日 144次阅读 来源: 月季卷心菜花

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))

测试结果：

《每天学点Python Cookbook（三）》

    原文作者：月季卷心菜花
    原文地址: https://segmentfault.com/a/1190000014784819
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞