Início / coding / Perguntas / 78679802
Accepted
ThomasIsCoding
Asked: 2024-06-28 05:04:06 +0800 CST

Como fazer uma implementação `functools.reduce` que se pareça com `Reduce` em R?

  • 772

Aqui está um exemplo R de usoReduce

x <- c(1, 2, 2, 4, 10, 5, 5, 7)
Reduce(\(a, b) if (tail(a, 1) != b) c(a, b) else a, x) # equivalent to `rle(x)$values`

O código acima serve para classificar os valores exclusivos da extração em termos de duração da execução, que podem ser facilmente obtidos por rle(x)$values.

Eu sei que em Python existe itertools.groupbyque executa a mesma coisa que rleem R, MAS, o que estou curioso é : é possível ter uma tradução altamente semelhante usando functools.reduceem Python para obter a mesma funcionalidade, digamos, por exemplo

from functools import reduce
x = [1,2,2,4,10,5,5,7]
reduce(lambda a, b: a + [b] if a[-1]!= b else a, x)

mas que infelizmente dá erros como

{
    "name": "TypeError",
    "message": "'int' object is not subscriptable",
    "stack": "---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[58], line 4
      1 from functools import reduce
      2 x = [1,2,2,4,10,5,5,7]
----> 4 reduce(lambda a, b: a + [b] if a[-1]!= b else a, x)

Cell In[58], line 4, in <lambda>(a, b)
      1 from functools import reduce
      2 x = [1,2,2,4,10,5,5,7]
----> 4 reduce(lambda a, b: a + [b] if a[-1]!= b else a, x)

TypeError: 'int' object is not subscriptable"
}

Minha pergunta é : existe alguma linha reduceem Python que se pareça com o código R?

2 respostas

  1. Best Answer

    você poderia usar uma lista como inicial:

    from functools import reduce
x = [1,2,2,4,10,5,5,7]

reduce(lambda a, b: a + [b] if a[-1]!= b else a, x, [x[0]])
[1, 2, 4, 10, 5, 7]

    Observe que você pode usar groupbyfrom itertools:

    
from itertools import groupby
[i for i,j in groupby(x)]
[1, 2, 4, 10, 5, 7]
    • 4

  2. Aqui está outra solução apenas para destacar a diferença. Em R, um escalar também é um vetor de comprimento 1, então tail() ou c() não tem problemas, enquanto Python não tem conceito de escalar[-1] ou escalar + lista.

    def foo(a,b):
  if (isinstance(a, int)): return [a]
  elif a[-1] != b:         return a + [b]
  else:                    return a

reduce(foo, x) 
[1, 2, 4, 10, 5, 7]
    • 2

relate perguntas