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

使用用户输入变量在列表中查找子字符串

聖光之護

发布： 2025-10-12 11:44:01

原创

964人浏览过

使用用户输入变量在列表中查找子字符串

<p>本文旨在指导开发者如何在一个包含多个元组的列表中，根据用户输入的字符串查找并返回匹配的元组。文章详细解释了原始代码存在的问题，并提供了多种解决方案，包括如何正确地在元组的字符串元素中查找子字符串，以及如何处理元组中可能存在的非字符串类型数据。同时，还介绍了如何仅匹配元组的特定部分（例如名称）。</p> ### 在列表的元组中查找子字符串在处理包含多个住宿地点的列表时，经常需要根据用户输入的关键词进行搜索。本文将介绍如何实现这一功能，并解决一些常见的问题。假设我们有以下列表，其中每个元素都是一个包含住宿地点信息的元组： ```python 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'), ]

我们的目标是让用户输入一个字符串，然后在列表中搜索包含该字符串的住宿地点信息，并打印结果。

初始尝试及问题分析

以下是最初的搜索函数：

def search_name():
    response = input()
    responses = [match for match in places if response in match]
    print(responses)

search_name()

登录后复制

这段代码的问题在于，response in match 检查的是 response 是否是整个 match (一个元组) 的元素，而不是检查 response 是否是元组中任何一个字符串元素的子字符串。例如，'Bec' in ('Becketts', 'Bed and Breakfast') 的结果是 False。

解决方案一：遍历元组元素

为了解决这个问题，我们需要遍历元组中的每个元素，并检查 response 是否是该元素的子字符串。

def search_name():
    response = input()
    responses = [match for match in places for item in match if response in item]
    print(responses)

search_name()

登录后复制

这段代码可以工作，但存在重复添加的问题。例如，当输入 'mantle' 时，Freemantle Solent Lodge 的数据会被添加两次，因为 'mantle' 在该元组的多个元素中都存在。

解决方案二：使用 any() 函数

为了避免重复添加，可以使用 any() 函数来检查 response 是否是元组中任何元素的子字符串。

def search_name():
    response = input()
    responses = [match for match in places if any(response in item for item in match)]
    print(responses)

search_name()

登录后复制

这段代码更有效率，因为它只循环 len(places) 次，并且对于每个元组，只要 response 是其中任何一个元素的子字符串，就会将该元组添加到结果中。

处理非字符串类型数据

如果元组中包含非字符串类型的数据，例如整数，直接使用 in 运算符会报错。例如，在示例数据中，Claremont Guest House 的元组包含一个字符串 '8'。如果需要处理整数类型，可以使用以下方法将其转换为字符串：

Alkaid.art

专门为Phtoshop打造的AIGC绘画插件

153

查看详情

def search_name():
    response = input()
    responses = [match for match in places if any(response in str(item) for item in match)]
    print(responses)

search_name()

登录后复制

这段代码将元组中的每个元素都转换为字符串，然后再进行子字符串匹配。

注意事项： 这种方法仅适用于所有值都可以转换为字符串的情况。

仅匹配名称

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

def search_name():
    response = input()
    responses = [(name, *_) for (name, *_) in places if response in name]
    print(responses)

search_name()

登录后复制

这段代码使用了解包操作符 *，将元组的第一个元素赋值给 name，其余元素赋值给 _。然后，它检查 response 是否是 name 的子字符串，并将整个元组添加到结果中。