[英]Return the first index of sublist if a given string is in any of the sublist
Market = [[1, 'apple', '45'], [2, 'banana', '76'], [3, 'apple', '67']
def search(data: List[list], search: str) -> List[int]:
"""
Return a list of IDs(first index) of fruits whose names contain search
"""
期望的输出:
>>> get_fruits_containing(Market, 'Apple')
[1, 3]
>>> get_bridges_containing(Market, 'bana')#part of name of fruit
[2]
"""
我尝试过(在市场中搜索 s 中的 s):但没有奏效。 它应该接受大写或小写。
尝试一个函数,如:
def get_fruits_containing(l,i):
return [x[0] for x in l if i.lower() in x[1]]
列表理解是你的朋友:-)。
现在您的查询正在运行:
>>> get_fruits_containing(Market, 'Apple') # Works with uppercase too.
[1, 3]
>>> get_fruits_containing(Market, 'bana')
[2]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.