stevensonmt

stevensonmt

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

Showing Posts 1 to 5

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

stevensonmt OP

Sorry, rust code should have used iter_mut() without a collect() call on a mutable vec. Good suggestion for the countdown accumulator. Thanks.

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

stevensonmt

stevensonmt OP

Thanks for all the help! Stream.concat is a pretty obvious. literal translation of what I was trying to do, so thanks for that tip. And simply pattern matching the index/position for applying the mapping function is probably the most obvious way to do what I want. Thanks again.

— All posts loaded —

Where Next? Top

Trending in Questions Top

Blokh
Hey guys, I’ve got a huge CSV ( around 10 GB ) that needs to be processed hourly Do you guys have any suggestions what is the best prac...
New
RSP87
I’m working on a project that simulates the bumbl example in the programming phoenix book. It acts almost like an email client. We have a...
New
kszambelanczyk
Hello! Could someone please give me a help/sample code, how to delete a file from s3 using waffle/waffle_ecto from Phoenix app. I creat...
New
RemyXRenard
I’m seeing that a list inside a Kino.DataTable will be interpreted as a charlist, even if the Kino.configure() is set to charlists: :as_l...
New
velrest
So my question is quite simple and i have found no conclusive answer on forum, google or AI. Should we use :erlang.float for Integer to ...
New
samoloth
Hi, I’ve just set up an application with ash_authentication. There is only magic link strategy for now, so there is no confirmation add o...
New
FlyingNoodle
If a change or preparation module uses Ash.Changeset.get_argument/2 or Ash.Query.get_argument/2 (or any of the other get_argument functio...
New

Other Trending Topics Top

mudasobwa
I am happy to introduce the very α version of the new programming language compiled to BEAM. Welcome Cure. It has literally three kille...
New
garrison
Hobbes is a low-level distributed database for the Elixir programming language. Hobbes provides a simple, safe, and scalable storage lay...
New
marciok
Hi there! We created Gust: A task orchestrator inspired by Airflow. For those who have never heard about Aiflow, it’s a Python-based wor...
New
jimsynz
Beam Bots (or just BB for short) is a framework for building fault-tolerant robotics applications in Elixir using familiar OTP patterns. ...
New
Damirados
Hello everyone. After busy few months I am happy to announce v0.1.0 of Emerge &amp; Solve. They are GUI (Emerge) and State management (S...
New
netoum
Corex is an accessible, unstyled UI component library for Phoenix that integrates Zag.js state machines using Vanilla JavaScript and Live...
New

We're in Beta

About us Mission Statement

Options

Thread Display Mode




Thread Preview

Skip Thread Previews