return a binary list elements between two lists based on a condition

user_10543310 0
Viewed 174

I have the following two lists:

A = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
B = ['b', 'd', 'f', 'g']

Of which I want to check if elements of B are in Athen 0 else 1. The expected output should be like this:

{'C': [0, 1, 0, 1, 0, 1, 1]}

I tried the following but didn't get what I am expecting.

{'C': [int(i == j) for j in A] for i in B}
list dictionary python
3 Answers

as @Boris mentioned using set(B) would be much faster if its a long list.

{'C': [int(value in set(B)) for value in A] }

output :

{'C': [0, 1, 0, 1, 0, 1, 1]}
user_132730540

As per your expected output (If A element is in B):

[1 if x in B else 0 for x in A]
user_43118390 
A = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
B = ['b', 'd', 'f', 'g']

C = [1 if a in B else 0 for a in A]

This is a one liner achieving what you are looking for. Nice and simple.

user_117475750
Related