[python] numpy运用

[cs_5050]

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