python - 列表输出中需要 float

Question

我正在尝试创建一个自定义过滤器，以使用 SciPy 包中的通用过滤器来运行它。

scipy.ndimage.filters.generic_filter

问题是我不知道如何使返回值成为标量，因为它需要通用函数才能工作。我通读了这些线程(底部)，但我找不到执行函数的方法。

代码是这样的:

import scipy.ndimage as sc

def minimum(window):    
    list = []
    for i in range(window.shape[0]):
        window[i] -= min(window)
        list.append(window[i])        
    return list

test =  np.ones((10, 10)) * np.arange(10)

result = sc.generic_filter(test, minimum, size=3)

它给出了错误:

cval, origins, extra_arguments, extra_keywords)
TypeError: a float is required

Scipy filter with multi-dimensional (or non-scalar) output

How to apply ndimage.generic_filter()

http://ilovesymposia.com/2014/06/24/a-clever-use-of-scipys-ndimage-generic_filter-for-n-dimensional-image-processing/

Answer 1

如果我理解的话，你想要将每个像素减去其 3 水平邻域的最小值。对列表执行此操作不是一个好习惯，因为 numpy 是为了提高效率(大约快 100 倍)。最简单的方法就是:

test-sc.generic_filter(test, np.min, size=3)

然后对整个数组进行减法向量化。您还可以这样做:

test-np.min([np.roll(test,1),np.roll(test,-1),test],axis=0)

如果您在边境接受了文物，速度会快 10 倍。