2017-10-18 5 views
2

Comment fusionner deux tableaux ensemble dans Elixir en utilisant la fonction Enum?Elixir - Fusionner 2 Liste des listes (comme des colonnes)

E.g.

points = [[1.0, 2.0], 
       [5.0, 3.0], 
       [2.0, 4.0], 
       [3.0, 1.0], 
       [2.0, 2.0], 
       [4.0, 3.0], 
       [1.0, 2.0], 
       [2.0, 5.0], 
       [14.0, 13.0], 
       [11.0, 12.0], 
       [12.0, 15.0], 
       [32.0, 25.0]] 

    results = [1.0, 
       3.0, 
       2.0, 
       1.0, 
       3.0, 
       2.0, 
       1.0, 
       3.0, 
       2.0, 
       1.0, 
       3.0, 
       2.0] 

Les tableaux combinés (colonne ajoutée), devrait ressembler à ceci:

combined = [[1.0, 2.0, 1.0], 
       [5.0, 3.0, 3.0], 
       [2.0, 4.0, 2.0], 
       [3.0, 1.0, 1.0], 
       [2.0, 2.0, 3.0], 
       [4.0, 3.0, 2.0], 
       [1.0, 2.0, 1.0], 
       [2.0, 5.0, 3.0], 
       [14.0, 13.0, 2.0], 
       [11.0, 12.0, 1.0], 
       [12.0, 15.0, 3.0], 
       [32.0, 25.0, 2.0]]  

Il semble que quelque chose comme ça devrait fonctionner:

points 
    |> Stream.with_index 
    |> Enum.map(fn({x,i}) -> Enum.into(x, Enum.fetch!(results, i)) end)     

Bien, cela produit une erreur:

** (Protocol.UndefinedError) protocol Collectable not implemented for 1.0. This protocol is implemented for: BitString, Ecto.Adapters.SQL.Stream, File.Stream, HashDict, HashSet, IO.Stream, List, Map, MapSet, Plug.Conn, Postgrex.Stream 
    (elixir) lib/collectable.ex:1: Collectable.impl_for!/1 
    (elixir) lib/collectable.ex:77: Collectable.into/1 
    (elixir) lib/enum.ex:1158: Enum.into_protocol/2 
    (elixir) lib/enum.ex:1259: anonymous fn/3 in Enum.map/2 
    (elixir) lib/enum.ex:1824: anonymous fn/3 in Enum.map/2 
    (elixir) lib/stream.ex:1413: anonymous fn/3 in Enumerable.Stream.reduce/3 
    (elixir) lib/stream.ex:978: anonymous fn/3 in Stream.with_index/2 
    (elixir) lib/enum.ex:3161: Enumerable.List.reduce/3 
    (elixir) lib/stream.ex:1433: Enumerable.Stream.do_each/4 
    (elixir) lib/enum.ex:1823: Enum.map/2 

Répondre

5

Chaque fois que vous voulez itérer sur deux enumerables en même temps, vous devriez penser à utiliser Enum.zip/2. Voilà comment je le ferais avec Enum.zip/2 et for:

for {point, result} <- Enum.zip(points, results) do 
    point ++ [result] 
end 
|> IO.inspect 

Sortie:

[[1.0, 2.0, 1.0], [5.0, 3.0, 3.0], [2.0, 4.0, 2.0], [3.0, 1.0, 1.0], 
[2.0, 2.0, 3.0], [4.0, 3.0, 2.0], [1.0, 2.0, 1.0], [2.0, 5.0, 3.0], 
[14.0, 13.0, 2.0], [11.0, 12.0, 1.0], [12.0, 15.0, 3.0], [32.0, 25.0, 2.0]] 

Re: Pourquoi votre code ne fonctionne pas: le deuxième argument de Enum.into devrait être un dénombrable. Vous voulez probablement l'envelopper dans une liste:

Enum.into(x, [Enum.fetch!(results, i)]) 

Je suggère d'utiliser la solution Enum.zip/2 bien.

+0

C'est génial, j'aime vraiment la réponse –

0

Vous pouvez utiliser le zip de Enum, comme:

resultado=for {point, res} <- Enum.zip(points, results) do 
    point ++ [res] 
    end