Question

我在R中使用data.tables工作。我有以下data.table编码一组带有坐标A，B，C，D的点和索引编码点所属的集合。

library(data.table)

      A B C D set
   1: 0 0 0 0   1
   2: 1 0 1 0   2
   3: 1 1 1 0   2
   4: 0 1 0 0   2
   5: 1 0 1 1   2
   6: 0 1 0 0   3
   7: 1 1 0 0   3
   8: 0 0 1 0   4
   9: 1 0 1 0   4
  10: 0 1 0 1   4
  11: 0 0 0 0   5
  12: 1 0 0 0   5
  13: 1 1 1 0   5
  14: 1 1 1 1   5

dt = setDT(structure(list(A = c(0L, 1L, 1L, 0L, 1L, 0L, 1L, 0L, 1L, 0L, 
0L, 1L, 1L, 1L), B = c(0L, 0L, 1L, 1L, 0L, 1L, 1L, 0L, 0L, 1L, 
0L, 0L, 1L, 1L), C = c(0L, 1L, 1L, 0L, 1L, 0L, 0L, 1L, 1L, 0L, 
0L, 0L, 1L, 1L), D = c(0L, 0L, 0L, 0L, 1L, 0L, 0L, 0L, 0L, 1L, 
0L, 0L, 0L, 1L), set = c(1L, 2L, 2L, 2L, 2L, 3L, 3L, 4L, 4L, 
4L, 5L, 5L, 5L, 5L)), .Names = c("A", "B", "C", "D", "set"), row.names = c(NA, 
-14L), class = "data.frame"))

我有另一个表编码，例如每组的概率。

   set       mass
1:   1 0.27809187
2:   2 0.02614841
3:   3 0.36890459
4:   4 0.28975265
5:   5 0.03710247

wt = setDT(structure(list(set = 1:5, mass = c(0.27809187, 0.02614841, 0.36890459, 
0.28975265, 0.03710247)), .Names = c("set", "mass"), row.names = c(NA, 
-5L), class = "data.frame"))

我想有一个程序来创建子空间的投影，例如光盘。（注意原始点1,4,6,7,11,12在这种情况下是一致的，在这个子空间中集合1和3是相同的，以及集合2和5。

unique(dt[,c("C","D", "set")])
>   C D set
 1: 0 0   1
 2: 1 0   2
 3: 0 0   2
 4: 1 1   2
 5: 0 0   3
 6: 1 0   4
 7: 0 1   4
 8: 0 0   5
 9: 1 0   5
10: 1 1   5

并识别相同的集合，保持唯一的集合并将相应的质量相加。即在这种情况下：

>   C D set
 1: 0 0   1
 2: 1 0   2
 3: 0 0   2
 4: 1 1   2
 5: 1 0   4
 6: 0 1   4

   set       mass
1:   1 0.6469965 % set 1 + set 3
2:   2 0.06325088 % set 2 + set 5
3:   4 0.36890459

感谢您的想法。

Answer 1

一个有点笨重的选项：为每个集合创建一个唯一的字符串，然后对其进行分组。

coords = c("C", "D")
gDT = setorder(unique(dt[,c(coords, "set"), with=FALSE]))[,
  .(s = paste(do.call(paste, c(.SD, .(sep="_"))), collapse="."))
, by=set, .SDcols = coords][, 
  g := .GRP
, by=s][]

#    set           s g
# 1:   1         0_0 1
# 2:   2 0_0.1_0.1_1 2
# 3:   3         0_0 1
# 4:   5 0_0.1_0.1_1 2
# 5:   4     0_1.1_0 3

gDT[wt, on=.(set), mass := i.mass ]
gDT[, .(set = first(set), mass = sum(mass)), by=g]

#    g set       mass
# 1: 1   1 0.64699646
# 2: 2   2 0.06325088
# 3: 3   4 0.28975265

您可以通过在最后一行链接g来摆脱[, g := NULL][]。
setorder只是对数据进行排序，以便唯一字符串在相同的集合集中相同。
已优化分组first和sum操作，因为您可以查看是否将verbose = TRUE添加到最后一行，例如gDT[, .(set = first(set), mass = sum(mass)), by=g, verbose=TRUE]。

Answer 2

在概念上类似于Frank，我们可以将每组的二进制值映射到带有coords = c("C", "D") d = data.frame(set = dt$set, val = Reduce("+", Map("*", list(dt$C, dt$D), 2 ^ ((length(coords) - 1):0)))) d的小数。同样，为＆＃34; C＆＃34;和＆＃34; D＆＃34;，我们得到：

tab = table(d$val, d$set) > 0L ## `table(d) > 0` to ignore the duplicates
gr = colSums(tab * (2 ^ ((nrow(tab) - 1):0)))
gr
# 1  2  3  4  5 
# 8 11  8  6 11

## another (pre-edit) alternative with unnecessary overhead
#gr = cutree(hclust(dist(table(d) > 0L)), h = 0)
#gr                        
#1 2 3 4 5 
#1 2 1 3 2

然后，我们可以按照相同的想法对相同的集合进行分组：

rowsum(wt$mass[match(names(gr), wt$set)], gr, reorder = FALSE)
#         [,1]
#8  0.64699646
#11 0.06325088
#6  0.28975265

并根据此群组进行汇总：

JSZip

R：data.table比较行集

2 个答案: