Getting the first elements per row in an array in Python?,zip(*s)[0]