numpy的获得不min或出三个元件

Question

的最大元件I具有三维数组：

y = np.random.randint(1,5 ,(50,50,3)) 

我想计算跨越第三轴线（3片）最大和最小，然后通过分剩下的数字/元素。

因此，像这样：

x = (np.max(y, axis =2) - 2*np.min(y, axis =2))/the third number 

我不知道如何得到第三个数字。 需要注意的是，第三个数字可能等于最小值或最大值：

例如， （5,5,1）

Answer 1

虽然通常排序是矫枉过正时，你只需要一个最大值和最小值，在这种情况下，我认为这是最简单的。它直接将我们想要的数字放在容易访问的地方，而不需要任何复杂的算术。

y = np.random.randint(1, 5, (50, 50,3)) 
y2 = y.copy() 
y2.sort(axis=2) 
sout = (y2[...,2] - 2 * y2[...,0])/y2[...,1] 

这给了我

In [68]: (sout == divakar_out).all() 
Out[68]: True 

这通常是一个好兆头。 ;-)

Answer 2

方法＃1

诀窍找到第三个从3减去那些最大和最小索引。当最大和最小索引相同时，即最后一个轴上的所有三个元素都是相同的，第三个元素索引也是相同的。

因此，我们有这样一个解决方案 -

max_idx = y.argmax(2) 
min_idx = y.argmin(2) 

rem_idx = np.where(max_idx == min_idx, max_idx, 3 - max_idx - min_idx) 
out = (y[all_idx(max_idx, 2)] -2*y[all_idx(min_idx, 2)])/y[all_idx(rem_idx, 2)] 

索引助手功能为y与指数 -

# https://stackoverflow.com/a/46103129/ @Divakar 
def all_idx(idx, axis): 
    grid = np.ogrid[tuple(map(slice, idx.shape))] 
    grid.insert(axis, idx) 
    return tuple(grid) 

方法2

我们可以得到沿着轴求和并减去最小值和最大值以得到第三个元素并简单插入公式 -

maxv = np.max(y, axis =2) 
minv = np.min(y, axis =2) 
x = (maxv - 2*minv)/(y.sum(2) - maxv - minv) 

Answer 3

另一种方法是使用np.median

(y.max(2) - 2 * y.min(2))/np.median(y, 2)