创建列,在其他列中给出下一个相等或更小的值

当前方法的主要问题是处理没有更低/相等值的情况——在这些情况下,我们需要到数据帧末尾的距离。你的前4应该是3,因为需要3行才能再次等于或小于1。

df1 |>
  mutate(
    col2 = sapply(seq_len(n()), \(x) {
      matches <- which(col1[-(1:x)] <= col1[x])
      if (length(matches)) matches[1] else n() - x + 1
    })
  )

我的方法是首先将行号作为显式行添加,因为您是在计算这些行号。然后,我进行滚动自连接,找到小于或等于当前行的下一行 col1 然后,我取行差或剩余行数之间的第一个NA。

library(dplyr)
df2 <- df1 |> mutate(row = row_number())
df2 |>
  left_join(df2, join_by(closest(row < row), col1 >= col1)) |>
  transmute(col1 = col1.x, col2 = coalesce(row.y - row.x, n() - row.x + 1))

结果

  col1 col2
1    1    3
2    2    1
3    2    1
4    1    4
5    2    2
6    3    1
7    2    1