『Python』获取列表中重复出现的某个元素的所有索引(index)

思路很简单…利用 D a t a F r a m e DataFrame DataFrame 中的index方法来实现

import pandas as pd
test = ['a', 'ca', 'aca', 2, 'a', 'b']

def get_multiple_index(key_word, lis):
    df = pd.DataFrame(lis, columns=['tmp'])
    res = df[df['tmp'] == key_word].index.tolist()
    # 元素出现一次以上
    if len(res) > 1:
        return res
    # 只出现一次
    elif len(res) == 1:
        return res[0]
    else:
        return None

print(get_multiple_index('a', test))
print(get_multiple_index('ca', test))
print(get_multiple_index('k', test))

"""
[0, 4]
1
None
"""