There are lots of solutions to do this for a single array, but what about a matrix, such as:
>>> k array([[ 35, 48, 63], [ 60, 77, 96],
nums = [[ 35, 48, 63], [ 60, 77, 96], [ 91, 112, 135]] highs = [max(lst) for lst in nums] highs[nth]