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

时间:2019-07-17 03:53:57

标签: numpy-ndarray

我手头有以下东西:

  
    具有形状(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循环来实现这一目标,但是我仍然想知道是否可以使用一些直接在专长上运行的切片技巧?

0 个答案:

没有答案
相关问题