63

我得到了一些简单的代码:

def find(str, ch):
    for ltr in str:
        if ltr == ch:
            return str.index(ltr)
find("ooottat", "o")

该函数仅返回第一个索引。如果我将返回更改为打印,它将打印 0 0 0。这是为什么,有什么办法可以得到0 1 2

4

12 回答 12

113

这是因为str.index(ch)将返回ch第一次出现的索引。尝试:

def find(s, ch):
    return [i for i, ltr in enumerate(s) if ltr == ch]

这将返回您需要的所有索引的列表。

PS Hugh 的回答显示了一个生成器函数(如果索引列表变大,它会有所不同)。此功能也可以通过更改[]为来调整()

于 2012-06-20T14:58:50.777 回答
23

我会选择 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)]
于 2012-06-20T15:19:33.130 回答
18

Lev 的答案是我会使用的答案,但是这里有一些基于您的原始代码的内容:

def find(str, ch):
    for i, ltr in enumerate(str):
        if ltr == ch:
            yield i

>>> list(find("ooottat", "o"))
[0, 1, 2]
于 2012-06-20T15:04:06.813 回答
11
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
于 2012-06-20T15:01:07.460 回答
3

在一行中获取所有位置

word = 'Hello'
to_find = 'l'

# in one line
print([i for i, x in enumerate(word) if x == to_find])
于 2021-06-03T15:05:14.040 回答
3
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]
于 2020-09-19T15:44:05.427 回答
1

根据经验,在使用 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
于 2020-02-21T11:22:51.603 回答
0

这是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
于 2020-04-08T00:09:41.927 回答
0

美化@Lev 和@Darkstar 发布的五星级单线:

word = 'Hello'
to_find = 'l'
print(", ".join([str(i) for i, x in enumerate(word) if x == to_find]))

这只是使索引号的分离更加明显。
结果将是: 2, 3

于 2021-07-19T22:15:57.663 回答
0
x = "abcdabcdabcd"
print(x)
l = -1
while True:
    l = x.find("a", l+1)
    if l == -1:
        break
    print(l)
于 2018-08-23T09:37:13.730 回答
0

所有其他答案都有两个主要缺陷:

  1. 他们通过字符串执行 Python 循环,这非常慢,或者
  2. 他们使用 numpy,这是一个非常大的附加依赖项。
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它不需要任何新的进口。

于 2020-09-09T18:00:35.817 回答
-1

你可以试试这个

def find(ch,string1):
    for i in range(len(string1)):
        if ch == string1[i]:
            pos.append(i)        
于 2013-05-03T07:13:41.063 回答