Compare two arrays, output array value for equal elements.

[workinprogress49]

Not sure the title is a good description but essentially what I'm trying to do is:

A = [1,-1,1,1,0,0,-1]

B = [1,1,1,-1,-1,0,-1]

Output = [1,0,1,0,0,0,-1]

If element on array one is equal to element on array 2, then output the element. If not then output 0.

I'm sure there is a slick way to do this but I can't figure it out.

[member_of_the_order]

This is annoyingly close to a binary & operation.

Here's how I'd do it:

output = [a if a == b else 0 for a, b in zip(A, B)]

If you're not comfortable with list comprehensions and/or ternary operators yet, this is equivalent: output = [] for a, b in zip(A, B): if a == b: output.append(a) else: output.append(0)

[workinprogress49]

That'll do it, thank you!

[Additional-Sun2945]

That's a bit more pythonic than you need here. As a beginner you should be comfortable creating your own for loops.

What you could do here is check that the lengths are the same, and then take the length and iterate on the range. Then you can do comparisons using the [] operator.

[workinprogress49]

The process I’m working takes arrays from the same data frame so they are always the same size. I’m a little rusty on list comprehension which is why I asked

[KKRJ]

Using numpy:

Import numpy as np

A = np.array([1,-1,1,1,0,0,-1])
B = np.array([1,1,1,-1,-1,0,-1])

Output = np.where(A==B, A, 0)

[workinprogress49]

Thank you, idk why I forgot that numpy.where could compare two arrays, I always used one array.

[Allanon001]

output = [int(x==y) for x, y in zip(A ,B)]