2017-10-18 88 views
2

如何在Elixir中使用Enum函数将两个数组合并在一起?Elixir - 合并2列表列表(如列)

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] 

的组合阵列(列加),应该是这样的:

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]]  

好像这样的事情应该工作:

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

虽然,这将产生一个错误:

** (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 

回答

5

每当你想同时遍历了两个可枚举,你应该考虑使用Enum.zip/2。这就是我如何与Enum.zip/2for做到这一点:

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

输出:

[[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]] 

回复:为什么您的代码不起作用:Enum.into的第二个参数应该是一个枚举。你可能想将其包装在一个列表:

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

我建议使用Enum.zip/2解决方案虽然。

+0

太棒了,我真的很喜欢答案 –

0

您可以使用zip从枚举,如:

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