2

谁能告诉我一个聪明的方法来计算稀疏矩阵中非零邻居的距离？这样做的原因是我想在一个大的多维矩阵中识别“洞”。

举个例子：假设我有一个 10x10 的矩阵，里面有很多零和一些 1：

f<-ceiling((runif(100,0,1))-.6)
a <- matrix(f, ncol=10L, nrow=10L)
rownames(a) = seq(1,10,1)
colnames(a) = seq(1,10,1)

一个可能是

> a
   1 2 3 4 5 6 7 8 9 10
1  0 0 0 0 1 0 0 1 1  0
2  1 0 0 0 1 0 0 0 0  0
3  0 1 1 0 1 1 0 0 1  1
4  0 0 1 1 0 1 0 1 0  0
5  0 0 0 1 0 1 0 0 0  1
6  1 0 0 0*0*1 0 1 0  0
7  0 1 1 1 1 1 1 0 1  0
8  1 0 1 0 1 0 0 0 1  1
9  1 1 1 0 0 0 0 0 0  0
10 1 1 1 1 1 0 1 0 1  0

因此，我需要一个 10x10 矩阵，其中所有零点的所有四个方向（每个维度两个）的平均距离。

例如，点a[6,5]为零并且有 4 个邻居。向左的距离是 5，向右是 1，向上是 3，向下是 1。所以平均值是 2,5。如果邻居不存在，则应计算剩余邻居的平均值。

像往常一样，我的第一个想法是一堆 for 循环，在各个方向上查找每个矩阵值并返回距离。但这一定是最愚蠢的做法......

4

2 回答 2

1

我非常喜欢这个问题！这是一个解决方案，不一定很快，但它可以完成工作。

首先让我们重新创建您的数据：

a <- matrix(scan(textConnection("
  0 0 0 0 1 0 0 1 1  0
  1 0 0 0 1 0 0 0 0  0
  0 1 1 0 1 1 0 0 1  1
  0 0 1 1 0 1 0 1 0  0
  0 0 0 1 0 1 0 0 0  1
  1 0 0 0 0 1 0 1 0  0
  0 1 1 1 1 1 1 0 1  0
  1 0 1 0 1 0 0 0 1  1
  1 1 1 0 0 0 0 0 0  0
  1 1 1 1 1 0 1 0 1  0
")), 10, 10, byrow = TRUE)

在这里，让我们将行和列拆分为四个面向向量的列表：

rev.list <- function(l) lapply(l, rev)

v1 <- split(a, row(a))  # rows left to right
v2 <- rev.list(v1)      # rows right to left
v3 <- split(a, col(a))  # cols up to down
v4 <- rev.list(v3)      # cols down to up

在这里，我们创建并应用一个函数（灵感来自https://stackoverflow.com/a/17929557/1201032）来计算单向距离：

dir.dist <- function(v) {
  out <- seq_along(v) - cummax(seq_along(v) * v)
  out[seq_len(match(1, v) - 1)] <- NA
  out
}

dist1.list <- lapply(v1, dir.dist) # dist to closest on left
dist2.list <- lapply(v2, dir.dist) # dist to closest on right
dist3.list <- lapply(v3, dir.dist) # dist to closest up
dist4.list <- lapply(v4, dir.dist) # dist to closest dn

现在让我们将所有内容放回四个矩阵中：

nr <- nrow(a)
nc <- ncol(a)

list.to.mat <- function(l, revert = FALSE, byrow = FALSE,
                           nrow = nr, ncol = nc) {
  x <- unlist(if (revert) rev.list(l) else l)
  matrix(x, nrow, ncol, byrow)
}

m1 <- list.to.mat(dist1.list, revert = FALSE, byrow = TRUE)
m2 <- list.to.mat(dist2.list, revert = TRUE,  byrow = TRUE)
m3 <- list.to.mat(dist3.list, revert = FALSE, byrow = FALSE)
m4 <- list.to.mat(dist4.list, revert = TRUE,  byrow = FALSE)

最后，让我们使用受https://stackoverflow.com/a/13123779/1201032pmean启发的函数计算均值：

pmean <- function(..., na.rm = FALSE) { 
  dat <- do.call(cbind, list(...))
  res <- rowMeans(dat, na.rm = na.rm) 
  idx_na <- !rowSums(!is.na(dat))
  res[idx_na] <- NA
  res 
}

final <- matrix(pmean(as.vector(m1),
                      as.vector(m2),
                      as.vector(m3),
                      as.vector(m4), na.rm = TRUE), nr, nc)

final
#       [,1] [,2] [,3] [,4]  [,5]  [,6] [,7] [,8] [,9] [,10]
#  [1,] 2.50 2.50 2.00 2.00  0.00  1.67 3.00  0.0 0.00 1.50
#  [2,] 0.00 1.67 1.67 2.00  0.00  1.00 3.50  2.0 2.00 3.00
#  [3,] 1.67 0.00 0.00 1.00  0.00  0.00 2.33  1.5 0.00 0.00
#  [4,] 2.00 1.67 0.00 0.00  1.50  0.00 1.67  0.0 1.67 1.33
#  [5,] 2.33 2.00 1.33 0.00  1.50  0.00 2.00  1.5 2.00 0.00
#  [6,] 0.00 2.25 2.00 1.75 *2.25* 0.00 1.00  0.0 1.67 1.67
#  [7,] 1.00 0.00 0.00 0.00  0.00  0.00 0.00  1.0 0.00 1.33
#  [8,] 0.00 1.00 0.00 1.25  0.00  1.67 1.75  2.0 0.00 0.00
#  [9,] 0.00 0.00 0.00 1.33  1.33  2.50 2.33  4.0 2.67 4.00
# [10,] 0.00 0.00 0.00 0.00  0.00  1.67 0.00  2.0 0.00 1.50
于 2013-08-02T01:00:00.817 回答
0

对于您的矩阵，您可以使用以下方法转换为稀疏 rxc 形式which：

rc <- which(a>0, arr.ind=TRUE)
rc[rc[,1] == 6, 2]   # 1, 6,8   candidates for "row-neighbors"
rc[rc[,2] == 5, 1]   # 1,2,3, 7,8,10  col-neighbors
于 2013-08-01T23:04:46.957 回答