Merge array at specific index in Ruby

Question

I need a method which returns a merged array starting a specified index. I've looked here, here and here without cracking it.

I can see that this concatenates but I want to update the array not simply combine them:

@a1 = [0,0,0,0,0]

a2 = [1,1]

def update_array(new_array)
  @a1.push(*new_array)  
end

update_array(a2)

I'd like the output to be something like:

#[0,1,1,0,0] or [0,0,0,1,1]

Depending on the index specified.

I haven't in this method because I'm not sure how to. The method should specify the index somewhere. — safikabis, Nov 25 '17 at 16:13

steenslag · Answer 1 · 2017-11-25T16:34:32.243

2

Arrays have a method which does that: insert

a1 = [0,0,0,0,0]
a2 = [1,1]

a1.insert(1, *a2) # => [0, 1, 1, 0, 0, 0, 0]

edited Nov 25 '17 at 16:34

answered Nov 25 '17 at 16:28

steenslag

79,051
16
138
171

I need to update each array element #=> [0, 1, 1, 0, 0] rather than just combine them – safikabis Nov 25 '17 at 16:56

score 2 · Accepted Answer · answered Nov 25 '17 at 17:12

You can use the normal element assignment, Array#[]= and pass in the start and length parameters:

Element Assignment — Sets the element at index, or replaces a subarray from the start index for length elements, or replaces a subarray specified by the range of indices.

(emphasis mine) So, for instance:

@a1 = [0,0,0,0,0]
a2 = [1,1]
@a1[1, a2.length] = a2
@a1 # => [0, 1, 1, 0, 0]
@a1 = [0,0,0,0,0]
@a1[@a1.length - a2.length, a2.length] = a2
@a1 # => [0, 0, 0, 1, 1]

Merge array at specific index in Ruby

2 Answers2