[python]numpy运用问题

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

'''
a 3*4，b 1*4
要求这这样的，b中每一个元素其索引表示a中的列，值表示a中的行
利用b快速从a中提取出相关元素，组成一个1*4的数组（用向量化的方法）
例：b中第一个元素2表示要提取a中第0列第2行的元素，即3
'''