根据一维数组从二维数组中取值

import numpy as np

a = np.array([[1,2],[3,4],[5,6],[7,8],[9,10]])

b = np.array([1,2,3,4,5])

print(a[b>3])