如何在data.frame中动态创建差异列或增量列？

Question

我的dataframe有来自Balance，Balance1，Balance2，...，Balance36的未结余额的列名。

我想为每个月之间的增量添加一列，即Delta2 = Balance2 - Balance1

如何通过以下方法简化。

dataset$delta1 = apply(dataset[, c("Balance1","Balance")], 1, function(x){x[2]-x[1]})
dataset$delta2 = apply(dataset[, c("Balance2","Balance1")], 1, function(x){x[2]-x[1]})
...
dataset$delta35 = apply(dataset[, c("Balance35","Balance34")], 1, function(x){x[2]-x[1]})
dataset$delta36 = apply(dataset[, c("Balance36","Balance35")], 1, function(x){x[2]-x[1]})

Answer 1

它归结为单线。首先，将数据集命名为short，df是通常的名称。然后，使用直接减法;没有必要调用apply()从另一列中减去一列：

df$delta1 <- df[,"Balance1"] - df[,"Balance"]
df$delta2 <- df[,"Balance2"] - df[,"Balance1"]
...
df$delta35 <- df[,"Balance35"] - df[,"Balance34")]
df$delta36 <- df[,"Balance36"] - df[,"Balance35")]

但由于整个计算具有规则结构，我们实际上只讨论生成Nx36差异数组，因此使用数字列索引。假设你的“Balance *”列索引是（50:85），你的delta_cols是100：135，或者其他什么。然后你的“平衡*”减法的LHS指数是balance_lhs <- (50:84)，RHS指数是(51:85)，或者只是((50:84)+1)（记住大多数操作员喜欢在R中添加矢量化）

所以你的Nx36阵列可以只通过单行生成：

df[,delta_cols] <- df[,(balance_lhs+1)] - df[,balance_lhs]

并且您可以通过编程方式计算delta_cols <- which(colnames(df) == c("delta1",...,"delta36")，以避免代码中的魔术数列索引。

Answer 2

使用lapply计算一行中所有36个比较的delta。

# Sample data (37 columns, labelled Balance, Balance1, ...)
set.seed(2017);
df <- as.data.frame(matrix(runif(37 * 100), ncol = 37));
colnames(df) <- paste("Balance", c("", seq(1:36)), sep = "");

# List of difference vectors (36 distance vectors, labelled delta1, ...)
lst <- lapply(2:ncol(df), function(i) df[, i] - df[, i - 1]);
names(lst) <- paste("delta", seq(1:36), sep = "");

# Combine with original dataframe
df <- cbind.data.frame(
    df,
    as.data.frame(lst));