cs_5050
cs_5050
2017-10-19 13:53

[python]numpy运用问题

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

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

1条回答