numpy 二维数组， 已知index，批量取值

前言

示例代码如下，numpy官网有介绍
反思：项目遇到的小问题， 一开始不知道怎么做，上网找了许久未解决。这种numpy操作问题应该先看numpy basic 介绍， 看基础操作中是否有你要的功能。 推荐查阅

1. 书籍：《Python for Data Analysis》
2. 官网：
   https://numpy.org/doc/stable/user/basics.html

import numpy as np
pairs = np.array([[0,1],[0,2],[0,3]])
# print(pairs[:,1:])
# print(np.ones(pairs.shape[0]).reshape(1,-1))

dist=np.array([[0, 0.1, 0.2, 0.3, 0.4],
               [0, 0.4, 0.3, 0.2, 0.1],
               [0, 0.4, 0.3, 0.2, 0.1]])

print('dist', dist)
idx1=pairs[:,0]
idx2=pairs[:,1]

print("idx1, idx2", idx1, idx2)
print(dist[idx1, idx2])

标签：index,pairs,取值,np,print,idx2,idx1,numpy
来源： https://blog.csdn.net/sinat_20276189/article/details/117368597