在Python列表中查找包含用户输入变量的子字符串-Python教程-PHP中文网

在Python列表中查找包含用户输入变量的子字符串

DDD

发布： 2025-10-12 10:14:12

原创

367人浏览过

在python列表中查找包含用户输入变量的子字符串

本文旨在解决如何在Python列表中查找包含用户输入子字符串的问题。我们将提供一个实际示例，展示如何使用列表推导式和`any()`函数高效地搜索包含特定字符串的元组，并讨论处理不同数据类型的注意事项，以及如何针对特定字段进行匹配。

列表子字符串搜索：高效方法

假设我们有一个包含住宿信息的列表，每个住宿信息以元组的形式存储，包含名称、类型和地址等信息。我们需要实现一个搜索功能，允许用户输入一个字符串，然后在列表中查找包含该字符串的住宿信息。

例如，我们有如下列表：

places = [
    ('Becketts', 'Bed and Breakfast', '11 Bellevue Terrace Southsea Portsmouth PO5 3AT'),
    ('Charles Hope Apartments', 'Apartment', 'Exchange Court Southampton SO14 3SB'),
    ('Claremont Guest House', 'Bed and Breakfast', '33-35 The Polygon Southampton SO15 2BP', '8'),
    ('Farmhouse Hotel', 'Hotel', 'Burrfields Rd Portsmouth PO3 5HH'),
    ('Ferry House Lodge', 'Bed and Breakfast', '472 Mile End Rd Portsmouth PO2 7BX'),
    ('Freemantle Solent Lodge', 'Bed and Breakfast', 'Park Rd Freemantle Southampton SO15 3BB'),
    ('Hammersmith Rooms', 'Hostel', '28-30 Bute Gardens London, W6 7DS'),
]

登录后复制

以下代码展示了一种实现搜索功能的有效方法：

立即学习“Python免费学习笔记（深入）”；

def search_name():
    response = input("请输入搜索关键词：")
    responses = [match for match in places if any(response in item for item in match)]
    print(responses)

search_name()

登录后复制

这段代码的核心在于列表推导式和any()函数的结合使用。

列表推导式: [match for match in places if any(response in item for item in match)] 遍历 places 列表中的每一个元组 match。
any() 函数: any(response in item for item in match) 检查元组 match 中的任何一个元素 item 是否包含用户输入的字符串 response。如果元组中至少有一个元素包含该字符串，则 any() 函数返回 True，该元组就被添加到结果列表 responses 中。

这种方法避免了不必要的循环，提高了搜索效率。

处理不同数据类型

在上面的例子中，住宿信息元组中的所有元素都是字符串。但是，如果元组中包含其他数据类型，例如整数，则需要进行类型转换，否则会引发 TypeError。

表单大师AI

一款基于自然语言处理技术的智能在线表单创建工具，可以帮助用户快速、高效地生成各类专业表单。

查看详情

例如，如果 places 列表包含一个整数值，如下所示：

places = [
    ('Becketts', 'Bed and Breakfast', '11 Bellevue Terrace Southsea Portsmouth PO5 3AT'),
    ('Claremont Guest House', 'Bed and Breakfast', '33-35 The Polygon Southampton SO15 2BP', 8), # 包含整数8
]

登录后复制

直接使用 response in item 会导致错误。为了解决这个问题，我们可以将 item 转换为字符串：

responses = [match for match in places if any(response in str(item) for item in match)]

登录后复制

str(item) 将元组中的每个元素转换为字符串，从而允许 in 运算符进行比较。注意： 这种方法仅适用于所有值都可以转换为字符串的情况。

仅匹配名称字段

如果只需要匹配住宿名称（元组的第一个元素），可以使用以下代码：

def search_name():
    response = input("请输入搜索关键词：")
    responses = [(name, *_) for (name, *_) in places if response in name]
    print(responses)

search_name()

登录后复制

这段代码使用了元组解包和 *args 的技巧。(name, *_) 将元组的第一个元素赋值给 name 变量，其余元素赋值给 _ 变量（通常用于表示不需要使用的变量）。然后，if response in name 只检查名称字段是否包含用户输入的字符串。最后，(name, *_) 确保结果列表中包含完整的元组。