Поиск того, начинается ли строка с одного из префиксов переменной длины списка

мне нужно выяснить, начинается ли имя с любого из префиксов списка, а затем удалить его, например:

if name[:2] in ["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_"]:
    name = name[2:]

вышеизложенное работает только для префиксов списка длиной два. Мне нужна такая же функциональность для переменной длины префиксов.

Как это сделать эффективно (мало кода и хорошей производительности)?

a для цикла итерации по каждому префиксу, а затем проверка name.startswith(prefix) наконец ломтик название по длине префикса работает, но это много кода, вероятно, неэффективно, и "не-весть".

у кого-нибудь есть хорошее решение?

10 ответов


немного трудно читать, но это работает:

name=name[len(filter(name.startswith,prefixes+[''])[0]):]

.startswith (префикс[, start[, end]])¶

Return True если строка начинается с префикса, в противном случае return Ложный. префикс также может быть кортежем префиксов для поиска. С необязательный запуск, тестовая строка, начинающаяся с этой позиции. С необязательный конец, прекратите сравнивать строку в этой позиции.

$ ipython
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
Type 'copyright', 'credits' or 'license' for more information
IPython 6.4.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: prefixes = ("i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_")

In [2]: 'test'.startswith(prefixes)
Out[2]: False

In [3]: 'i_'.startswith(prefixes)
Out[3]: True

In [4]: 'd_a'.startswith(prefixes)
Out[4]: True

for prefix in prefixes:
    if name.startswith(prefix):
        name=name[len(prefix):]
        break

Если вы определяете префикс как символы перед подчеркиванием, то вы можете проверить

if name.partition("_")[0] in ["i", "c", "m", "l", "d", "t", "e", "b", "foo"] and name.partition("_")[1] == "_":
    name = name.partition("_")[2]

Regexes, вероятно, даст вам лучшую скорость:

prefixes = ["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_", "also_longer_"]
re_prefixes = "|".join(re.escape(p) for p in prefixes)

m = re.match(re_prefixes, my_string)
if m:
    my_string = my_string[m.end()-m.start():]

как насчет использования filter?

prefs = ["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_"]
name = list(filter(lambda item: not any(item.startswith(prefix) for prefix in prefs), name))

обратите внимание, что сравнение каждого элемента списка с префиксами эффективно останавливается при первом совпадении. Это поведение гарантируется any функция, которая возвращается, как только она находит True значением, например:

def gen():
    print("yielding False")
    yield False
    print("yielding True")
    yield True
    print("yielding False again")
    yield False

>>> any(gen()) # last two lines of gen() are not performed
yielding False
yielding True
True

или re.match вместо startswith:

import re
patt = '|'.join(["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_"])
name = list(filter(lambda item: not re.match(patt, item), name))

когда дело доходит до поиска и эффективности всегда думает о методах индексирования, чтобы улучшить свои алгоритмы. Если у вас есть длинный список префиксов, вы можете использовать индекс в памяти, просто индексируя префиксы первым символом в dict.

это решение стоит только если вас был длинный список префиксов и производительность становится проблемой.

pref = ["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_"]

#indexing prefixes in a dict. Do this only once.
d = dict()
for x in pref:
        if not x[0] in d:
                d[x[0]] = list()
        d[x[0]].append(x)


name = "c_abcdf"

#lookup in d to only check elements with the same first character.
result = filter(lambda x: name.startswith(x),\
                        [] if name[0] not in d else d[name[0]])
print result

регулярное выражение, испытано:

import re

def make_multi_prefix_matcher(prefixes):
    regex_text = "|".join(re.escape(p) for p in prefixes)
    print repr(regex_text)
    return re.compile(regex_text).match

pfxs = "x ya foobar foo a|b z.".split()
names = "xenon yadda yeti food foob foobarre foo a|b a b z.yx zebra".split()

matcher = make_multi_prefix_matcher(pfxs)
for name in names:
    m = matcher(name)
    if not m:
        print repr(name), "no match"
        continue
    n = m.end()
    print repr(name), n, repr(name[n:])

выход:

'x|ya|foobar|foo|a\|b|z\.'
'xenon' 1 'enon'
'yadda' 2 'dda'
'yeti' no match
'food' 3 'd'
'foob' 3 'b'
'foobarre' 6 're'
'foo' 3 ''
'a|b' 3 ''
'a' no match
'b' no match
'z.yx' 2 'yx'
'zebra' no match

это редактирует список на лету, удаляя префиксы. The break пропускает остальные префиксы, как только один найден для определенного элемента.

items = ['this', 'that', 'i_blah', 'joe_cool', 'what_this']
prefixes = ['i_', 'c_', 'a_', 'joe_', 'mark_']

for i,item in enumerate(items):
    for p in prefixes:
        if item.startswith(p):
            items[i] = item[len(p):]
            break

print items

выход

['this', 'that', 'blah', 'cool', 'what_this']

import re

def make_multi_prefix_replacer(prefixes):
    if isinstance(prefixes,str):
        prefixes = prefixes.split()
    prefixes.sort(key = len, reverse=True)
    pat = r'\b(%s)' % "|".join(map(re.escape, prefixes))
    print 'regex patern :',repr(pat),'\n'
    def suber(x, reg = re.compile(pat)):
        return reg.sub('',x)
    return suber



pfxs = "x ya foobar yaku foo a|b z."
replacer = make_multi_prefix_replacer(pfxs)               

names = "xenon yadda yeti yakute food foob foobarre foo a|b a b z.yx zebra".split()
for name in names:
    print repr(name),'\n',repr(replacer(name)),'\n'

ss = 'the yakute xenon is a|bcdf in the barfoobaratu foobarii'
print '\n',repr(ss),'\n',repr(replacer(ss)),'\n'