我得到了一些简单的代码:
def find(str, ch):
for ltr in str:
if ltr == ch:
return str.index(ltr)
find("ooottat", "o")
该函数仅返回第一个索引。如果我将返回更改为打印,它将打印 0 0 0。这是为什么,有什么办法可以得到0 1 2
?
这是因为str.index(ch)
将返回ch
第一次出现的索引。尝试:
def find(s, ch):
return [i for i, ltr in enumerate(s) if ltr == ch]
这将返回您需要的所有索引的列表。
PS Hugh 的回答显示了一个生成器函数(如果索引列表变大,它会有所不同)。此功能也可以通过更改[]
为来调整()
。
我会选择 Lev,但值得指出的是,如果您最终进行更复杂的搜索,那么使用 re.finditer 可能值得牢记(但 re 往往带来的麻烦多于价值 - 但有时很容易知道)
test = "ooottat"
[ (i.start(), i.end()) for i in re.finditer('o', test)]
# [(0, 1), (1, 2), (2, 3)]
[ (i.start(), i.end()) for i in re.finditer('o+', test)]
# [(0, 3)]
Lev 的答案是我会使用的答案,但是这里有一些基于您的原始代码的内容:
def find(str, ch):
for i, ltr in enumerate(str):
if ltr == ch:
yield i
>>> list(find("ooottat", "o"))
[0, 1, 2]
def find_offsets(haystack, needle):
"""
Find the start of all (possibly-overlapping) instances of needle in haystack
"""
offs = -1
while True:
offs = haystack.find(needle, offs+1)
if offs == -1:
break
else:
yield offs
for offs in find_offsets("ooottat", "o"):
print offs
结果是
0
1
2
在一行中获取所有位置
word = 'Hello'
to_find = 'l'
# in one line
print([i for i, x in enumerate(word) if x == to_find])
def find_idx(str, ch):
yield [i for i, c in enumerate(str) if c == ch]
for idx in find_idx('babak karchini is a beginner in python ', 'i'):
print(idx)
输出:
[11, 13, 15, 23, 29]
根据经验,在使用 POD(Plain Old Data)时,NumPy 数组的性能通常优于其他解决方案。字符串是 POD 的一个例子,也是一个字符。要查找字符串中只有一个字符的所有索引,NumPy ndarrays 可能是最快的方法:
def find1(str, ch):
# 0.100 seconds for 1MB str
npbuf = np.frombuffer(str, dtype=np.uint8) # Reinterpret str as a char buffer
return np.where(npbuf == ord(ch)) # Find indices with numpy
def find2(str, ch):
# 0.920 seconds for 1MB str
return [i for i, c in enumerate(str) if c == ch] # Find indices with python
这是Mark Ransom答案的略微修改版本,如果ch
长度可能超过一个字符,则可以使用。
def find(term, ch):
"""Find all places with ch in str
"""
for i in range(len(term)):
if term[i:i + len(ch)] == ch:
yield i
美化@Lev 和@Darkstar 发布的五星级单线:
word = 'Hello'
to_find = 'l'
print(", ".join([str(i) for i, x in enumerate(word) if x == to_find]))
这只是使索引号的分离更加明显。
结果将是: 2, 3
x = "abcdabcdabcd"
print(x)
l = -1
while True:
l = x.find("a", l+1)
if l == -1:
break
print(l)
所有其他答案都有两个主要缺陷:
def findall(haystack, needle):
idx = -1
while True:
idx = haystack.find(needle, idx+1)
if idx == -1:
break
yield idx
这通过haystack
查找进行迭代needle
,总是从上一次迭代结束的地方开始。它使用比逐个字符str.find
迭代快得多的内置函数。haystack
它不需要任何新的进口。
你可以试试这个
def find(ch,string1):
for i in range(len(string1)):
if ch == string1[i]:
pos.append(i)