stevensonmt

stevensonmt

Using Enum or Stream to pass through first n elements and map the rest

I’m used to Rust where iterators have a ton of handy functions. What I’d like to do is equivalent to the following Rust code:

my_iterator.iter()
           .skip(n)
           .map(|x| my_fun(x))
           .collect()

In more concrete terms

fn my_fun(x: i32) -> i32 { x * 10 }
let my_iterator = vec![1,2,3,4,5];
my_iterator.iter()
           .skip(2)
           .map(|x| my_fun(x))
           .collect::<Vec<i32>>()
// [1,2,30,40,50]

Is there an equivalent in Elixir? Or would I need to do something like

my_list = [1,2,3,4,5]
map_or_not(my_list)

def map_or_not([1 | [ 2 | t ]]) do
  [1 | [2 |map_or_not(t)]]
end 

def map_or_not(list) do 
  list |> Enum.map(fn x -> x * 10 end)
end

Most Liked

brettbeatty

brettbeatty

That’s not the behavior I’m seeing from skip. When I run that I get [30, 40, 50]. It appears to behave like Enum.drop/2 or Stream.drop/2.

As for how to get the behavior you want, you could use Enum.reduce/3 or Stream.scan/3, using an acc to count down elements not to touch, then performing the mapping function when the acc is 0.

Or even easier might be Enumerable.map_reduce/3:

{result, _acc} =
  Enum.map_reduce([1, 2, 3, 4, 5], 2, fn
    item, 0 ->
      {item * 10, 0}

    item, acc ->
      {item, acc - 1}
  end)

result
# => [1, 2, 30, 40, 50]
al2o3cr

al2o3cr

You should be able to write exactly what you mean:

s = 1..5

new_stream =
  Stream.concat(
    Stream.take(s, 2),
    Stream.drop(s, 2) |> Stream.map(& &1*10)
  )

Enum.to_list(new_stream) # => [1, 2, 30, 40, 50]

One caveat: this will traverse the first two elements of the stream twice. That’s no problem for most Elixir data structures, but will cause strange behavior when streaming from a stateful resource like the result of File.stream!.

ondrej-tucek

ondrej-tucek

I think you can make it simply

list = 1..5 # or list = [1,2,3,4,5]

skip_by_val = fn
   x, y when x <= y -> x
   x, _y -> 10*x
end

list |> Stream.map(&skip_by_val.(&1, 2)) |>  Enum.to_list()

or

skip_by_idx = fn
   {x, idx_x}, idx when idx_x <= idx -> x
   {x, _idx_x}, _idx -> 10*x   
end

list |> Stream.with_index(1) |> Stream.map(&skip_by_idx.(&1, 2)) |> Enum.to_list()

Where Next?

Popular in Questions Top

yawaramin
In the Dialyzer docs ( http://erlang.org/doc/man/dialyzer.html#requesting-or-suppressing-warnings-in-source-files ), there is a way to tu...
New
LegitStack
I’m hoping you guys can give me some general advice and perhaps code examples if you’re feeling up to it. I’m very interested in Elixir,...
New
jerry
Good day to you all. I have been struggling to get a query involving like and ilike to work. Can anyone assist me on this, please? pro...
New
sergio_101
I am VERY much an elixir newbie. I have taken one elixir course and one phoenix course on Udemy. During that course, I saw the instructor...
New
tduccuong
Hi, is there any work on GUI with Elixir, that is similar to Electron/Javascript? My idea is to bundle Phoenix and BEAM into a single se...
New
quazar
How to set Jason to encode all fields in ecto schema, I don’t care about security and implementing only is taking long list of attributes...
New
mcarvalho
What is the difference between System.get_env and Application.get_env? For example, what are best practices to use one versus another.
New
aadeshere1
I have a another noob question about loop. Since elixir is immutable, while loop is not directly possible. total = 10 while total != 0 ...
New
shahryarjb
Hello, I have map which I want to convert it to string like this: the map: %{last_name: "tavakkoli", name: "shahryar"} the string I ne...
New
vrod
I am using the Starship cross-shell prompt – it seems pretty nice, but I get some errors: [WARN] - (starship::utils): Executing command ...
New

Other popular topics Top

TunkShif
This post is an instruction guide to help you setup your Neovim for Elixir development from scratch. It includes general information on h...
273 38985 115
New
belgoros
I’m not a pro in using Regex and can’t figure out why the following behaviour happens, especially if we take into account the difference ...
New
freewebwithme
Using vs code and installed ElixirLS: support and debugger. And I got an error popped up on start up says Failed to run ‘elixir’ comma...
New
srinivasu
How to handle excepions in elixir? Suppose i have A, B, C ,D, E modules. and each module has get() function. A.get() method will call th...
New
William
I would like to know that is there any online source for learning Phoenix Framework for building E-Commerce Store? Any advantage on build...
New
yawaramin
In the Dialyzer docs ( http://erlang.org/doc/man/dialyzer.html#requesting-or-suppressing-warnings-in-source-files ), there is a way to tu...
New
JorisKok
I have a server on AWS, and was running a load test using artillery. When looking at the Phoenix dashboard I see the Ports going to 100% ...
New
aalberti333
As the title describes, I’m trying to run Enum.map() over a list of key/value pairs, where the value is a map. My data looks like this: ...
New
joeerl
Hello again - after a longish gap I’ve decided I really must dig into Elixir and see what’s been happening here - so I have a few questio...
New
magnetic
Hey :wave:t3: Elixir community, I’ve been learning Elixir, and working on some side projects. My editor of choice is VSCode, and althoug...
New

We're in Beta

About us Mission Statement