【think python】判断一组 `key: str` 中是否存在一个 key `in` a_string?

RDpWTeHM
  • 58

#think-python#

判断一组 key: str 中是否存在一个 key in a_string?是否有更好的办法?

【需求】

判断「一组」key 中,是否存在一个 key 在(in) string 中?

【当前实现】

mapping = {'合租': '合租',
           '整租': '整租', }

data = [
    '合租(4室1厅2卫)',
    '合租(2室1厅1卫)',
    '整租(3室1厅)',
    '整租(2室1厅1卫)',
]


def reclassify(s: str):
    for key in mapping:
        if key in s:
            return mapping[key]
    return s

for i in range(len(data)):
    data[i] = reclassify(data[i])

【问题】

主要是 line 13~line 16 有更好的写法吗?

总感觉效率有点低(时间复杂度略高)

评论
阅读 390
2 个回答
for k in mapping:
    for i in range(len(data)):
        if k in data[i]:
            data[i] = k

这种写法未必效率低。提供另一种使用正则的思路吧

import re
pattern = re.compile('|'.join(['({})'.format(i) for i in mapping]))

for i in range(len(data)):
    search_result = re.search(pattern, data[i])
    if search_result:
        data[i] = search_result.group()

无非就是判断一个字符串是否为另一个字符串的子串
你可以这样,

  • in 操作触发魔术方法 __contains__(self, key, /)
  • 用字符串的实例方法 find(...)
search = '合租'
target = '合租(4室1厅2卫)'
print(search in target)  # True

search='整租'
print(target.find(search) !=-1) # False
撰写回答

登录后参与交流、获取后续更新提醒

宣传栏