How to find two specific values in a large 6-column array in Python