Início / coding / Perguntas / 79021907
Accepted
DaniCee
Asked: 2024-09-25 16:49:57 +0800 CST

Aplicar função a uma lista dependendo de valores em outra lista

  • 772

Provavelmente é uma duplicata, mas não consigo encontrar esse caso específico.

Digamos que eu tenha duas listas:

list1 <- as.list(c(5, 8, 9))
list2 <- as.list(c(8, 10, 11))

Eu sei como aplicar uma função a uma lista. Digamos que eu queira aplicar o seguinte a list1:

lapply(list1, function(x){x+100})

Mas e se eu só quiser aplicar tal função list1quando o valor correspondente em list2for >=10? Então meu resultado é:

[[1]]
[1] 5

[[2]]
[1] 108

[[3]]
[1] 109

E mais ainda, como posso somar 100 ao valor em list1e subtrair o valor em list2somente quando o valor em list2for >=10?

Algo como o seguinte, que obviamente não funciona:

lapply(list1, list2, function(x,y){if (y>=10) x+100-y})

Obrigado!

3 respostas

  1. Best Answer

    Acho que você está procurando por Map(), que é mapply()sem simplificação:

    mapplyé uma versão multivariada de sapply. mapplyaplica-se FUNaos primeiros elementos de cada ...argumento, aos segundos elementos, aos terceiros elementos e assim por diante.

    Por razões que imagino que sejam conhecidas apenas pelos desenvolvedores de R, ao contrário de lapply(), sapply()e tapply(), a função é o primeiro argumento de mapply()e Map().

    Map(
    \(x1, x2) if (x2 >= 10) x1 + 100 else x1,
    list1,
    list2
)
# [[1]]
# [1] 5

# [[2]]
# [1] 108

# [[3]]
# [1] 109

    Se você quiser ser mais sofisticado, pode remover a if()declaração e definir a função como, \(x1, x2) x1 + 100 * (x2 >= 10)mas, pessoalmente, acho que a falta de explicitude a torna menos compreensível.

    Claro que você também pode fazer isso ifelse(list2 >= 10, unlist(list1) + 100, list1)neste caso, mas Map()é mais geral.

    Seu segundo exemplo poderia ser expresso como Map(\(x, y) if (y >= 10) x + 100 - y else x, list1, list2).

    • 3

  2. Dado o exemplo do brinquedo, não vejo por que usamos um formato de lista. As coisas ficam mais fáceis se trabalharmos com vetores em vez disso:

    list1 <- as.list(c(5, 8, 9))
list2 <- as.list(c(8, 10, 11))
x1 = unlist(list1)
x2 = unlist(list2)

    e então aplicar um vetor totalmente

    x1 + 100 * (x2 >= 10)
# [1]   5 108 109

    Nota. Notação retirada da solução do usuário @SamR .

    • 2

  3. Uma variante de usoMap

    > Map(`+`, list1, 100 * (unlist(list2) >= 10))
[[1]]
[1] 5

[[2]]
[1] 108

[[3]]
[1] 109
    • 1

relate perguntas