[elixir! #0083] Stream.transform 的用法

在 elixir 里可以用 Stream 来表示无限长的序列,例如 0,1,2,3... 就可以表示为:

iex> s = Stream.iterate(0, & &1 + 1)
#Function<62.50989570/2 in Stream.unfold/2>

如果我们想要计算这个数列中每5个数的和,就可以使用 Stream.transform 函数:

iex> s1 = Stream.transform(s, {0, 0}, fn x, {sum, count} ->
...> if count == 5 do                                      
...> {[sum], {x, 1}}                                       
...> else
...> {[], {sum + x, count + 1}}                            
...> end
...> end)
#Function<60.50989570/2 in Stream.transform/3>
iex> Enum.take(s1, 10)
[10, 35, 60, 85, 110, 135, 160, 185, 210, 235]

它是 Enum.flat_map_reduce 的 Stream 版本。

你可能感兴趣的