如何将任意形状的2D ndarray转换为1D ndarray

Question

我手头有以下东西：

  

    具有形状（n，d）的

2D ndarray（n个对象的特征），例如说n = 5，d = 3

  

feat = np.random.randn(5, 3)

  

    具有形状（K，2）的

2D ndarray（K个感兴趣的对象对），其中每一行是两个感兴趣的对象（例如，对象）的索引的组合。说K = 3

  

interest = np.array([[1, 2], [3, 4], [2, 0]])

  

    

具有（$ R ^ 3 $，$ R ^ 3 $）映射的函数f ----> $ R ^ 1 $

  

我想要的是形状为（K，1）的一维ndarray，其中第i个元素由以下项定义：

f(feat[interest[i][0]], feat[interest[i][1]]).

我知道我可以编写一个for循环来实现这一目标，但是我仍然想知道是否可以使用一些直接在专长上运行的切片技巧？