有条件的列表理解

Question

I have the following solution, however its wasteful to loop over the data again to remove pointless white-space (" ") placed in to satisfy the compiler. I wish to keep this solution compact and in-line as shown.

ans = [x[0] if x[1]==minimum else " " for x in zip(a,b)]

Answer 1

List comphrensions support the inclusion of a predicate.

I think you want:

ans = [x[0] for x in zip(a,b) if x[1]==minimum]

or maybe a little clearer like this:

ans = [x for (x,y) in zip(a,b) if y==minimum]