提取array中指定的某几行

import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])

print(a)
i=[2,0]#在这里将i按顺序赋值为所需要的行号即可，比如本例中想要依次输出第2行和第0行
print(a[i,:])

输出为