Проверить массив - это значение, содержащееся в другом массиве?

Question

Я хотел бы вернуть логическое значение для каждого значения в массиве A, которое указывает, находится ли он в массиве B. Это должна быть стандартная процедура, я думаю, но я не могу найти никакой информации о том, как это сделать. Моя попытка ниже:

A = ['User0','User1','User2','User3','User4','User0','User1','User2','User3' 
    'User4','User0','User1','User2','User3','User4','User0','User1','User2' 
    'User3','User4','User0','User1','User2','User3','User4','User0','User1' 
    'User2','User3','User4','User0','User1'] 
B = ['User3', 'User2', 'User4'] 
contained = (A in B) 

Однако я получаю ошибку:

ValueError: shape mismatch: objects cannot be broadcast to a single shape 

Я использую NumPy так любое решение, используя NumPy или стандартный Python будет предпочтительным.

Answer 1

Вы можете использовать in1d Я считаю -

np.in1d(A,B) 

Answer 2

Для тестирования без использования numpy, попробуйте:

contained = [a in B for a in A] 

результат:

[False, False, True, True, True, False, False, True, False, False, 
False, True, True, True, False, False, False, True, False, False, 
True, True, True, False, False, True, True, False, False]