繁体   English   中英

返回连续值满足条件的行

[英]Return rows where consecutive values meet criterion

我有以下数据帧df 我想返回一个向量result ,指示哪些行符合以下标准:该行中至少有2个连续值低于-1.7。

set.seed(123)

df <- data.frame(V1=rnorm(10,-1.5,.5),
                 V2=rnorm(10,-1.5,.5),
                 V3=rnorm(10,-1.5,.5),
                 V4=rnorm(10,-1.5,.5),
                 V5=rnorm(10,-1.5,.5),
                 V6=rnorm(10,-1.5,.5),
                 V7=rnorm(10,-1.5,.5),
                 V8=rnorm(10,-1.5,.5),
                 V9=rnorm(10,-1.5,.5),
                 V10=rnorm(10,-1.5,.5))
rownames(df) <- c(seq(1976,1985,1))

结果将是一个向量:

result <- c(1977,1979,1980,1982,1983,1985)

一个选项是使用apply遍历行,使用rle创建逻辑条件,检查是否有any lengths大于1的TRUE元素,提取names

names(which(apply(df, 1, function(x) with(rle(x < - 1.7), any(lengths[values] > 1)))))
#[1] "1977" "1979" "1980" "1982" "1983" "1985"

或者更好的方法是通过放置两个逻辑矩阵来对其进行矢量化(即删除数据集的第一列,检查它是否小于-1.7,同样删除最后一列并执行相同操作), ReduceReduce为单个逻辑matrix通过检查相应元素是否为TRUE ,获取rowSums ,如果值大于0,则提取行名称

names(which(rowSums(Reduce(`&`, list(df[-ncol(df)] < -1.7, df[-1] < -1.7))) > 0))
#[1] "1977" "1979" "1980" "1982" "1983" "1985"

一个有趣的选项which使用arr.ind = TRUE

temp <- which(df < -1.7, arr.ind = TRUE)
rownames(df)[aggregate(col~row, temp, function(x) any(diff(x) == 1))[, 2]]

#[1] "1977" "1979" "1980" "1982" "1983" "1985"

我们首先获得值小于-1.7的所有行和列位置。 使用aggregate我们为每一row组合col ,并检查row是否至少有一个连续值,以及返回其rownames TRUE子集的值。

一种解决方案,它使用滞后和来获得向量中每对数字的总和。 如果滞后总和为2,则表示该行中至少有2个连续值满足条件。

rownames(df)[apply(df < -1.7, 1, function(x) any(x[-nrow(df)] + x[-1] == 2))]

# [1] "1977" "1979" "1980" "1982" "1983" "1985"

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM