Home > Software engineering >  Show adjacent members in a list
Show adjacent members in a list

Time:04-22

I want to inspect adjacent elements in a list based on a match. For example, in a list of randomly ordered letters, I want to know what the neighboring letters of m is. My current solution is:

library(stringr)
ltrs <- sample(letters)
ltrs[(str_which(ltrs,'m')-2):(str_which(ltrs,'m') 2)]
[1] "j" "f" "m" "q" "a"

To me, the repetition of str_which() feels unnecessary. Is there a simpler way to achieve the same result?

CodePudding user response:

First, I regenerate random data with a seed for reproducibility:

set.seed(42)
ltrs <- sample(letters)
ltrs
#  [1] "q" "e" "a" "j" "d" "r" "z" "o" "g" "v" "i" "y" "n" "t" "w" "b" "c" "p" "x"
# [20] "l" "m" "s" "u" "h" "f" "k"

Use -2:2 and then (cautionarily) remove those below 1 or above the length of the vector:

ind <- -2:2   which(ltrs == "m")
ind <- ind[0 < ind & ind < length(ltrs)]
ltrs[ind]
# [1] "x" "l" "m" "s" "u"

If your target is more than one (not just "m"), then we can use a different approach.

ind <- which(ltrs %in% c("m", "f"))
ind <- lapply(ind, function(z) { z <- z   -2:2; z[0 < z & z <= length(ltrs)]; })
ind
# [[1]]
# [1] 19 20 21 22 23
# [[2]]
# [1] 23 24 25 26
lapply(ind, function(z) ltrs[z])
# [[1]]
# [1] "x" "l" "m" "s" "u"
# [[2]]
# [1] "u" "h" "f" "k"

Or, if you don't care about keeping them grouped, we can try this:

ind <- which(ltrs %in% c("m", "f"))
ind <- unique(sort(outer(-2:2, ind, ` `)))
ind <- ind[0 < ind & ind <= length(ltrs)]
ltrs[ind]
# [1] "x" "l" "m" "s" "u" "h" "f" "k"

CodePudding user response:

If you don't have duplicates, you can try the code like below

ltrs[seq_along(ltrs)%in% (which(ltrs=="m") (-2:2))]

otherwise

ltrs[seq_along(ltrs) %in% c(outer(which(ltrs == "m"), -2:2, ` `))]

CodePudding user response:

You can also use the slider::slide function (using data provided by @r2evans):

slider::slide(ltrs, ~ .x, .before = 2, .after = 2)[[which(ltrs == "m")]]
# [1] "x" "l" "m" "s" "u"

slider::slide(ltrs, ~ .x, .before = 2, .after = 2)[which(ltrs %in% c("m","f"))]
# [[1]]
# [1] "x" "l" "m" "s" "u"
#
# [[2]]
# [1] "u" "h" "f" "k"
  • Related