查找文件，返回查找特定文字中前n个匹配到的行

最新推荐文章于 2023-01-16 23:09:14 发布

原创最新推荐文章于 2023-01-16 23:09:14 发布 · 574 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

78 篇文章

订阅专栏

本文介绍了一种利用Python标准库中的Deque数据结构实现文件搜索的方法。该方法可以在遍历文件内容的同时，记录最近找到的匹配项，并且能够限制保存的匹配项数量。通过一个具体的例子展示了如何使用Deque来优化搜索效率。

from collections import deque  # 一个很实用的队列。
def search(lines, search_data, max_len=5):
    find_data_list = deque(maxlen=max_len)#设置函数里面最多保存的数量
    for i in lines:
        if search_data in i:
            find_data_list.append(search_data)
            yield i,find_data_list #使用yield返回数据。
            if len(find_data_list) == 5:
                return
path = "/home/guo/PycharmProjects/hellotest/tst.csv"
with open(path) as f:
    for searchline, search_data_list in search(f, "16:00", 4):
        for i in search_data_list:
            print(i, end="******")
        print(searchline)
        print("-" * 30)