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
'''
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
'''