我有一个稀疏的3D数组值。我试图转动每个"点"通过对数组应用高斯滤波器,进入模糊的球体#34;
我希望点(x,y,z)处的原始值保持不变。我只是想围绕这一点创建衰减值......但是应用高斯滤波器也会改变原始(x,y,z)值。
我目前正在这样做:
dataCube = scipy.ndimage.filters.gaussian_filter(dataCube, 3, truncate=8)
有没有办法让我对此进行规范化,或者做一些事情以使我的原始值仍然在这个新的dataCube中?如果这不是最好的方法,我不一定要使用高斯滤波器。
答案 0 :(得分:2)
您可以使用带有1作为其中心值的内核的卷积来执行此操作,并且宽度小于数据点之间的间距。
一维示例:
import numpy as np
import scipy.signal
data = np.array([0,0,0,0,0,5,0,0,0,0,0])
kernel = np.array([0.5,1,0.5])
scipy.signal.convolve(data, kernel, mode="same")
给出
array([ 0. , 0. , 0. , 0. , 2.5, 5. , 2.5, 0. , 0. , 0. , 0. ])
请注意,对于大型数组,fftconvolve可能要快得多。您还必须指定在数组边界应该发生的事情。
更新:三维示例
import numpy as np
from scipy import signal
# first build the smoothing kernel
sigma = 1.0 # width of kernel
x = np.arange(-3,4,1) # coordinate arrays -- make sure they contain 0!
y = np.arange(-3,4,1)
z = np.arange(-3,4,1)
xx, yy, zz = np.meshgrid(x,y,z)
kernel = np.exp(-(xx**2 + yy**2 + zz**2)/(2*sigma**2))
# apply to sample data
data = np.zeros((11,11,11))
data[5,5,5] = 5.
filtered = signal.convolve(data, kernel, mode="same")
# check output
print filtered[:,5,5]
给出
[ 0. 0. 0.05554498 0.67667642 3.0326533 5. 3.0326533
0.67667642 0.05554498 0. 0. ]