r 子集并从 data.frame 中的其他列中减去一列

Question

我想知道是否有人可以帮助我编写一个函数来执行此操作。我有一个如下所示的数据框：

df <- data.frame(TestA1 = c(1,2), TestA2 = c(2,3), TestB1 = c(3,4), TestB2 = c(5,6), A = c(1,1), B = c(2,2))
df
TestA1 TestA2 TestB1 TestB2 A B
1      2      3      5 1 2
2      3      4      6 1 2

我想从名称中包含 A 的所有其他列（即 TestA1 和 TestA2）中减去“A”列，并对 B 及其匹配列执行相同操作。在一个功能中尝试和做这件事太多了吗？

我还是 R 新手，我一直在我的数据集上逐个进行此操作，它似乎非常低效（完整数据集为 25000x55）。我一直在努力思考如何编写一个函数来做到这一点，并且一直在苦苦挣扎。任何帮助/指导将不胜感激！

编辑

理想的输出是（如果选择 B）

NewTestB1  NewTestB2
1          3
2          4

因此，B = c(2,2) 将从匹配的每一列中减去，从而生成一个输出数据帧，其中包含已选择的列和额外的调整。

Answer 1

在base R中，提供你想要的功能：

myF <- function( object, selector )
{
    ix <- which( grepl( selector, colnames( object ) ) & colnames( object ) != selector )
    rowSums( object[ ix ] ) - object[ selector ]
}

所以打电话

myF( df, "B" ) 

会给你

  B
1 6
2 8

这是基于您肯定简化的示例，可能需要进行一些调整。您可能还想重命名返回的 data.frame，但这一切都取决于您对结果的进一步使用。

如果您想返回一个向量，而不是一列 data.frame，请更改为

rowSums( object[ ix ] ) - object[ selector ][ , 1 ]

编辑：

这可能就是你想要的：

myF <- function( object, selector )
{
    ix <- which( grepl( selector, colnames( object ) ) & colnames( object ) != selector )
    for( i in ix )
        object[ length( object ) + 1 ] <- object[ i ] - object[ selector ]
    return( object )
}

这给了你

myF( df, "B" )
  TestA1 TestA2 TestB1 TestB2 A B TestB1.1 TestB2.1
1      1      2      3      5 1 2        1        3
2      2      3      4      6 1 2        2        4

可能会进行一些改进，例如更好的列名等，但功能已实现。另外，请注意，这不会更改原始 data.frame，但会为您提供更新的副本。

第二次编辑，删除列：

myF <- function( object, selector )
{
    ix <- which( grepl( selector, colnames( object ) ) & colnames( object ) != selector )
    for( i in ix )
        object[ length( object ) + 1 ] <- object[ i ] - object[ selector ]
    object <- object[ -ix ]
    object <- object[ -(which( colnames( object ) == selector ) ) ]
    return( object )
}

生产

myF( df, "B" )
  TestA1 TestA2 A TestB1.1 TestB2.1
1      1      2 1        1        3
2      2      3 1        2        4