r - R dplyr如何提取基于窗口的特征

Question

我想知道我该如何做这件事：

features_windows <- function(data, window_size) {
  data_with_feature <- data %>%
    mutate(window_one = nrow(data[data$ID<ID & data$key==key & data$ID > ID - window_size,]))
   data_with_feature
}

基本上，新功能应该是时间窗口 ID-window_size 中有多少点：带有密钥的 ID。目前我到处都是 0，for 循环在 R 中是一个糟糕的主意。有什么想法吗？

问候，T

Answer 1

找到了解决方案：

在开头使用 group_by 运算符：

features_windows <- function(data) {
  data_with_feature <- data %>%
    group_by(1:n()) %>%
    mutate(window_1 = nrow(data[data$ID < ID & data$ID > (ID - 10) & data$key == key, ]) ) %>%
  data_with_feature