列表理解双循环 if

Question

I have two lists A and B where B is a superset of A .

I would like boolean list of that specifies whether an element of A is equal to an element of B . The superset list B is ordered.

If have tried this:

res = [(1 for a in A if a is b) for b in B]

According to other answers this should work. But in my case it returns a list of generators.. How can I end up with just a normal list. Something like:

A = ['a', 'b', 'c', 'e']
B = ['a', 'b', 'c', 'd', 'e', 'f']
res = [1, 1, 1, 0, 1, 0]

Answer 1

The reason you get generator objects back is because you use () , hence those make generator expressions, which are not evaluated until you iterate through them.

What you really should be using if simple in operator. Example -

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

---

If your usecase is more complex and you have to use is operator to compare elements of A and B . Then you can use any() function to iterate through the generator expression and return True if a match is found, other False . Example -

res = [1 if any(a is b for a in A) else 0 for b in B]

Demo -

>>> A = ['a', 'b', 'c', 'e']
>>> B = ['a', 'b', 'c', 'd', 'e', 'f']
>>> res = [1 if any(a is b for a in A) else 0 for b in B]
>>> res
[1, 1, 1, 0, 1, 0]

---

Also, according to your question -

I would like boolean list of that specifies whether an element of A is equal to an element of B.

If what you really want is a boolean (rather than 1 or 0 ) , then you can simply change above examples to -

Example 1 -

res = [x in A for x in B]

Example 2 -

res = [any(a is b for a in A) for b in B]

Answer 2

I think what you're looking for is the intersection of the two sets, right?

Here's the intersection of the two sets (which I highly recommend you come up with better names for next time):

A = [1, 2, 3, 5, 7, 10, 12, 9]

B = [2, 6, 7]

print [x if x in B else None for x in A]

You need the "else" in there or it doesn't work.

And to solve your problem literally:

 res = [True if x in A else False for x in B]

I recommend you check this answer out: if/else in Python's list comprehension?