我的函数不断检查相同的索引而不是检查另一个索引

Question

def capital_indexes(word):
    letters = []

    capital_letters = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']

    index_letters = []

    # append letters in the word to a list
    for letter in word:
        letters.append(letter)

    # go through the letters in the list letters
    for i in letters:
        # checks if the indexes in the list match with the indexes in the list 'capital_leters'
        if i in capital_letters:
            index_letters.append(word.index(i))

    return index_letters


word_capital = capital_indexes('TEsT')
# this should return (0, 1, 3) but it keeps returning (0, 1, 0)

print(word_capital)

Answer 1

.index 返回第一个出现的字母，使用enumerate 和.isupper：

def capital_indexes(word):
    return [i for i,l in enumerate(word) if l.isupper()]

Answer 2

您的代码效率非常低，例如转换为列表是无用的（您可以遍历字符串），然后循环遍历所有大写字母也是低效的（而是使用集合）。最后，您的主要错误在于 word.index(i) 将始终返回 first 匹配的位置（因此 0 代替 3）。

这是一个替代解决方案：

def capital_indexes(word):
    # general case for any given set of characters
    from string import ascii_uppercase
    letters = set(ascii_uppercase)
    return [i for i,c in enumerate(word) if c in letters]
    # case for uppercase only
    # return [i for i,c in enumerate(word) if c.isupper()]
        
capital_indexes('TEsT')
# [0, 1, 3]

Answer 3

这是因为 T 在索引 0 和索引 3 处，所以索引 0 被返回两次。

试试：

import string

result = [i for i, char in enumerate(word) if char in string.ascii_uppercase]

在你的例子中：

>>> word = "TEsT"
>>> [i for i, char in enumerate(word) if char in string.ascii_uppercase]

[0, 1, 3]