Как вставить новый элемент между всеми элементами массива Руби?

у меня есть массив и я хочу вставить новый элемент между всеми элементами, так или иначе, как join метод. Например, у меня есть

[1, [], "333"]

и то, что мне нужно-это

[1, {}, [], {}, "333"]

Примечание новый пустой хэш был вставлен между всеми элементами.

изменить: В настоящее время у меня есть:

irb(main):028:0> a = [1, [], "333"]
=> [1, [], "333"]
irb(main):029:0> a = a.inject([]){|x, y| x << y; x << {}; x}
=> [1, {}, [], {}, "333", {}]
irb(main):030:0> a.pop
=> {}
irb(main):031:0> a
=> [1, {}, [], {}, "333"]
irb(main):032:0>

Я хочу знать, лучший способ.

8 ответов


[1, 2, 3].flat_map { |x| [x, :a] }[0...-1]
#=> [1, :a, 2, :a, 3]

FYI, эта функция называется вкрапить (по крайней мере в Haskell).

[Update] если вы хотите избежать среза (который создал копию массива):

[1, 2, 3].flat_map { |x| [x, :a] }.tap(&:pop)
#=> [1, :a, 2, :a, 3]

a = [1,2,3]
h, *t = a
r = [h]
t.each do |e|
  r.push({}, e)
end
r #=> [1, {}, 2, {}, 3]

вы могли бы сделать что-то вроде:

a = [1, [], "333"]
new_a = a.collect {|e| [e, {}]}.flatten(1)
=> [1, {}, [], {}, "333", {}]

вам нужно .flatten(1) потому что он сгладит ваш пустой массив без него.

или, как говорит @ David Grayson в комментарии, Вы можете сделать flat_map который будет делать то же самое.

a.flat_map {|e| [e, {}]}
=> [1, {}, [], {}, "333", {}]

@tokland имеет правильный ответ, если последний {} не надо. Вы возвращаете срез от 0 до длины-1 или [0..-1].


другое подобное решение использует #product :

[1, 2, 3].product([{}]).flatten(1)[0...-1]
# => [ 1, {}, 2, {}, 3 ]

irb(main):054:0* [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(1).flat_map {|e| e << "XXX"}[0...-1]
=> [1, "XXX", 2, "XXX", 3, "XXX", 4, "XXX", 5, "XXX", 6, "XXX", 7, "XXX", 8, "XXX", 9]
irb(main):055:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(2).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, "XXX", 3, 4, "XXX", 5, 6, "XXX", 7, 8, "XXX", 9]
irb(main):056:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(3).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, "XXX", 4, 5, 6, "XXX", 7, 8, 9]
irb(main):057:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(4).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, "XXX", 5, 6, 7, 8, "XXX", 9]
irb(main):058:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(5).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, 5, "XXX", 6, 7, 8, 9]
irb(main):059:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(6).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, 5, 6, "XXX", 7, 8, 9]
irb(main):060:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(7).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, 5, 6, 7, "XXX", 8, 9]
irb(main):061:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(8).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, 5, 6, 7, 8, "XXX", 9]
irb(main):062:0> [1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(9).flat_map {|e| e << "XXX"}[0...-1]
=> [1, 2, 3, 4, 5, 6, 7, 8, 9]
irb(main):063:0>

еще один, который похож на Tokland это:

xs.inject([]){|x,y| x << y << {}}[0...-1]

один подход состоит в том, чтобы zip другой массив желаемых элементов, а затем сгладить его с depth = 1:

> arr = [1, [], "333"]
> element = {}
> interspersed = arr.zip([element] * (arr.size - 1)).flatten(1).compact
> # [1, {}, [], {}, "333" ]

вы можете продлить Array сделать это поведение более доступными.

class Array
  def intersperse(elem)
    self.zip([elem] * (self.size - 1)).flatten(1).compact
  end
end

например,

[43] pry(main)> [1,2,3].intersperse ('a')
=> [1, "a", 2," a", 3]


[1, 2, 3, 4, 5].inject { |memo, el| Array(memo) << {} << el }
#=> [1, {}, 2, {}, 3, {}, 4, {}, 5]

inject будет использовать первый элемент для начала, поэтому вам не нужно возиться с индексами.