Question

我有一个大型数据集，我将其拆分为子集。对于每个子集，我必须进行相同的计算，但使用不同的数字。例如：

Main Table
x a b c d
A 1 2 4 5 
A 4 5 1 7
A 3 5 6 2
B 4 5 2 9
B 3 5 2 8
C 4 2 5 2
C 1 9 6 9
C 1 2 3 4
C 6 3 6 2

 Additional Table for A
  a b c d
A 5 1 6 1

Additional Table for B
  a b c d
B 1 5 2 6

Additional Table for C
  a c c d
C 8 2 4 1

我需要将主表中的所有行A与来自附加表的A，主表中的所有行B以及来自B的值和主表中所有行B的值相乘，其值为C.如果这使得解决方案更容易，那么将其他表合并为一个完全没问题。

我考虑过for循环，但我无法将不同的乘数（来自附加表）放入代码中。由于存在大量子组，因此应避免手动编码每个乘法。我该如何进行乘法运算？

Answer 1

我们可以在Map主数据'df'之后使用split（假设所有数据集都是data.frame s。

df[-1] <- unsplit(Map(function(x,y) x*y[col(x)], 
                     split(df[-1], df$x),
                     list(unlist(dfA), unlist(dfB), unlist(dfC))), df$x)
df
#  x  a  b  c  d
#1 A  5  2 24  5
#2 A 20  5  6  7
#3 A 15  5 36  2
#4 B  4 25  4 54
#5 B  3 25  4 48
#6 C 32  4 20  2
#7 C  8 18 24  9
#8 C  8  4 12  4
#9 C 48  6 24  2

或者我们可以使用data.table

更快的选项

library(data.table)
setnames(setDT(do.call(rbind, list(dfA, dfB, dfC)), keep.rownames=TRUE)[df, 
 .(a= a*i.a, b= b*i.b, c = c*i.c, d= d*i.d),  on = c('rn' = 'x'), by = .EACHI], 1, 'x')[]
#   x  a  b  c  d
#1: A  5  2 24  5
#2: A 20  5  6  7
#3: A 15  5 36  2
#4: B  4 25  4 54
#5: B  3 25  4 48
#6: C 32  4 20  2
#7: C  8 18 24  9
#8: C  8  4 12  4
#9: C 48  6 24  2

如果有很多列，则上述情况会很困难，在这种情况下，我们可以使用mget检索列，并使用{{1}在*列上执行i. }

Map

Answer 2

如果我们从加法表开始为addDf，主表开始为df：

addDf
  x a b c d
A A 5 1 6 1
B B 1 5 2 6
C C 8 2 4 1

我们可以使用矩阵的合并和按元素乘法，

df[-1] <- merge(addDf, data.frame(x = df[1]), by = "x")[-1] * df[order(df[1]), -1]
df
  x  a  b  c  d
1 A  5  2 24  5
2 A 20  5  6  7
3 A 15  5 36  2
4 B  4 25  4 54
5 B  3 25  4 48
6 C 32  4 20  2
7 C  8 18 24  9
8 C  8  4 12  4
9 C 48  6 24  2

注意：从@akrun借用一点语法糖作为df[-1]赋值。

R中不同子集的不同数据的乘法

2 个答案: