三维二进制数组中到其他类型最近点的距离

Question

我有一个三维numpy数组，由1和0组成，定义了多孔固体中的开放空间和填充空间（目前是numpyint64数组）。我想确定从“1”点（体素）到最近的零点的欧几里得距离。有简单的方法吗？在

Answer 1

如果val包含值（0或1），并且pos包含每个体素的位置，则可以使用scipy.spatial.distance.cdist计算所有成对距离：

import numpy as np
from scipy.spatial.distance import cdist
# Find the points corresponding to zeros and ones
zero_indices = (val == 0)
one_indices = (val == 1)
# Compute all pairwise distances between zero-points and one-points
pairwise_distances = distance.cdist(pos[zero_indices, :], pos[one_indices, :])
# Choose the minimum distance
min_dist = np.min(pairwise_distances, axis=0)