Python文本文件特定词汇检索：图书管理程序搜索功能求助

阿华AIGC实验室

2026-5-19

解决Python图书编目程序的搜索功能问题

Hey there! Let's work through getting that title/author search feature up and running for your book catalog program. I've been in similar spots before, so let's break this down step by step.

第一步：规范文本文件的存储格式

首先建议你用统一的分隔符来存储每本图书的信息，比如每行存 书名|作者，这样后续读取和解析会非常方便。举个例子，你的books.txt内容可以是：

百年孤独|加西亚·马尔克斯
Python编程：从入门到实践|埃里克·马瑟斯
小王子|安托万·德·圣埃克苏佩里

这种格式避免了因空格、换行等混乱导致的搜索匹配失败，是文本存储小型数据的常用方式。

第二步：实现搜索功能的核心逻辑

接下来我们可以写一个搜索函数，先让用户选择搜索类型（书名/作者），然后输入关键词，再遍历文本文件的每一行进行匹配。这里要注意两个实用细节：

大小写不敏感匹配（比如用户搜“python”也能找到“Python编程...”）
支持部分匹配（比如搜“马尔克斯”就能找到对应的书）

完整代码示例

def add_book():
    """添加图书到文本文件"""
    title = input("请输入书名：")
    author = input("请输入作者：")
    with open("books.txt", "a", encoding="utf-8") as f:
        f.write(f"{title}|{author}\n")
    print("图书添加成功！")

def list_books():
    """列出全部图书（后续可添加排序逻辑）"""
    try:
        with open("books.txt", "r", encoding="utf-8") as f:
            books = f.readlines()
        if not books:
            print("当前暂无图书记录")
            return
        print("=== 全部图书 ===")
        for idx, book in enumerate(books, 1):
            title, author = book.strip().split("|")
            print(f"{idx}. 书名：{title}，作者：{author}")
    except FileNotFoundError:
        print("当前暂无图书记录")

def search_books():
    """按书名或作者搜索图书"""
    print("请选择搜索类型：")
    print("1. 按书名搜索")
    print("2. 按作者搜索")
    choice = input("输入选项（1/2）：")
    
    if choice not in ["1", "2"]:
        print("无效选项，请重新选择")
        return
    
    keyword = input("请输入搜索关键词：").lower()
    found = False
    
    try:
        with open("books.txt", "r", encoding="utf-8") as f:
            books = f.readlines()
        
        print("\n=== 搜索结果 ===")
        for book in books:
            book = book.strip()
            if not book:
                continue
            title, author = book.split("|")
            # 根据选择的类型进行匹配
            if (choice == "1" and keyword in title.lower()) or (choice == "2" and keyword in author.lower()):
                print(f"书名：{title}，作者：{author}")
                found = True
        
        if not found:
            print("未找到匹配的图书")
    except FileNotFoundError:
        print("当前暂无图书记录")

def main():
    while True:
        print("\n=== 图书编目程序 ===")
        print("1. 添加图书")
        print("2. 列出全部图书")
        print("3. 搜索图书")
        print("4. 退出程序")
        choice = input("输入选项（1/2/3/4）：")
        
        if choice == "1":
            add_book()
        elif choice == "2":
            list_books()
        elif choice == "3":
            search_books()
        elif choice == "4":
            print("程序退出")
            break
        else:
            print("无效选项，请重新输入")

if __name__ == "__main__":
    main()