У меня есть список:
['a','b','b','c']
чтобы найти все вхождения элемента, который я использую:
incd=['a','b','b','c']
indeces=[i for i, x in enumerate(incd) if x == 'b']
как я могу искать два элемента и все их позиции?
w1='a'
w2='b'
indeces=[i for i, x in enumerate(incd) if x == w1|w2]
возвращается
TypeError: unsupported operand type(s) for |: 'str' and 'str'
а также
indeces=[i for i, x in enumerate(incd) if x == 'a|b']
возвращается
[]
оба терпят неудачу
Я хотел бы вернуться
[0, 1, 2]






вы делаете это: вы должны использовать условие «или»
incd = ['a', 'b', 'b', 'c']
w1 = 'a'
w2 = 'b'
indeces = [i for i, x in enumerate(incd) if x == w1 or x== w2]
если у вас много данных для тестирования: используйте список
w = ['a', 'b' ,'d',...]
indeces = [i for i, x in enumerate(incd) if x in w]
а если перейти на w7? Это будет длинное заявление
Предлагаю пройти через Операторы в Python.
Замените это:
if x == w1|w2
С этим:
if x == w1 or x == w2
перечислите список и проверьте, равен ли элемент w1 или w2:
s = ['a','b','b','c']
w1 = 'a'
w2 = 'b'
for indx, elem in enumerate(s):
if elem == w1 or elem == w2:
print("Elem: {} at Index {}".format(elem, indx))
ВЫХОД:
Elem: a at Index 0
Elem: b at Index 1
Elem: b at Index 2
Более короткая версия:
print([i for i, e in enumerate(s) if e == w1 or e == w2]) # to have a tuple of both elem and indx replace i with (e,i)
ВЫХОД:
[0, 1, 2]
люблю объяснение
@alex Спасибо! Удачного кодирования! :)
indeces=[i for i, x in enumerate(incd) if x == w1 or x == w2]
Привет, пожалуйста, добавьте объяснение, почему это решение. Спасибо
Использование set для большей скорости и динамического количества элементов для поиска:
find = {2, 3} # The elements we want to find
a = [1,2,2,3,4] # our list
x = [ind for ind, val in enumerate(a) if val in find]
print(x)
МИУК,
s=pd.Series(incd)
s[s.eq(w1)|s.eq(w2)].index
#Int64Index([0, 1, 2], dtype='int64')
Вы можете использовать оператор in для выражения логического отношения ИЛИ.
incd = list('abcd')
w1, w2 = 'a', 'b'
indeces = [i for i, x in enumerate(incd) if x in [w1, w2]]
Это приводит к правильным индексам по желанию
indeces = [0, 1]
Поскольку вы отмечаете панд
l=['a','b','b','c']
s=pd.Series(range(len(l)),index=l)
s.get(['a','b'])
Out[893]:
a 0
b 1
b 2
dtype: int64
Используйте defaultdict со списком по умолчанию
from collections import defaultdict
d = defaultdict(list)
for n,e in enumerate(incd):
d[e].append(n)
Что в d в этот момент?
>>> d
defaultdict(<class 'list'>, {'a': [0], 'b': [1, 2], 'c': [3]})
и получить позиции «a» или «b» (и доказать, что это работает для ключа 'foo' не в incd)
print( d['a']+d['b']+d['foo'] )
# gives [0,1,2]
Почему здесь отмечен
pandas?