Поиск того, начинается ли строка с одного из префиксов переменной длины списка
мне нужно выяснить, начинается ли имя с любого из префиксов списка, а затем удалить его, например:
if name[:2] in ["i_", "c_", "m_", "l_", "d_", "t_", "e_", "b_"]:
name = name[2:]
вышеизложенное работает только для префиксов списка длиной два. Мне нужна такая же функциональность для переменной длины префиксов.
Как это сделать эффективно (мало кода и хорошей производительности)?
a для цикла итерации по каждому префиксу, а затем проверка name.startswith(prefix)
наконец ломтик название по длине префикса работает, но это много кода, вероятно, неэффективно, и "не-весть".
у кого-нибудь есть хорошее решение?
10 ответов
.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
Если вы определяете префикс как символы перед подчеркиванием, то вы можете проверить
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'