Могу ли я создать массив в Ruby со значениями по умолчанию?

Question

Perl довольно хорош в значениях по умолчанию:

: jmglov@laurana; perl -e '@foo; printf "%d\n", $foo[123]'
0
: jmglov@laurana; perl -e '%foo; printf "%d\n", $foo{bar}'
0

Ruby может сделать то же самое, по крайней мере для хэшей:

>> foo = Hash.new(0)
=> {}
>> foo[:bar]
=> 0

Но то же самое, похоже, не работает для массивов:

>> foo = Array.new(0)
=> []
>> foo[123]
=> nil
>> foo[124] = 0
=> 0
>> foo[456] = 0
=> 0
>> foo[455,456]
=> [nil, 0]

Можно ли указать значение по умолчанию для массивов, поэтому, когда они автоматически расширяются, они заполняются 0 вместо нуля?

Конечно, я могу обойти это, но ценой выразительности:

>> foo[457,458] = 890, 321
=> [890, 321]
>> foo[456] += 789
NoMethodError: You have a nil object when you didn't expect it!
You might have expected an instance of Array.
The error occurred while evaluating nil.+
>> foo.inject(0) {|sum, i| sum += (i || 0) }
=> 1211
>> foo.inject(:+)
NoMethodError: You have a nil object when you didn't expect it!
You might have expected an instance of Array.
The error occurred while evaluating nil.+

Обновление 1: Один из моих коллег отметил, что я могу использовать #compact для решения проблемы #inject и #to_i для решения стандартной проблемы "элемент-индекс":

>> foo.include? nil
=> true
>> foo.compact.inject(:+)
=> 1211
>> foo[456,457]
=> [0, 890, 321]
>> foo[455..457]
=> [nil, 0, 890]
>> foo[455..457].map(&:to_i)
=> [0, 0, 890]

Обновление 2: Спасибо Эндрю Гримму за решение проблемы +=:

>> foo = []
=> []
>> def foo.[](i)
>>   fetch(i) {0}
>> end
=> nil
>> foo[4]
=> 0
>> foo
=> []
>> foo[4] += 123
=> 123
>> foo
=> [nil, nil, nil, nil, 123]

Обновление 3: это начинает выглядеть как хер-моль!

>> foo
=> [nil, nil, nil, nil, 123]
>> foo[-2..-1]
TypeError: can't convert Range into Integer

Но мы можем справиться с этим:

>> def foo.[](index)
>>   if index.is_a? Range
>>     index.map {|i| self[i] }
>>   else
?>     fetch(index) { 0 }  # default to 0 if no element at index; will not cause auto-extension of array
>>   end
>> end
=> nil
>> foo
=> [nil, nil, nil, nil, 123]
>> foo[-2..-1]
=> [nil, 123]

Теперь я должен (смущенно) признать, что подкласс Array, чтобы не загромождать мой код:

class MyClass
  class ArrayWithDefault < Array
    def [](index)
      if index.is_a? Range
        index.map {|i| self[i] }
      else
        fetch(index) { 0 }  # default to 0 if no element at index; will not cause auto-extension of array
      end
    end
  end
end

Спасибо за все креативные решения. TIMTOWTDI действительно!

Answer 1

Не расширяется автоматически, но инициализируется до указанной длины со значением по умолчанию:

>> Array.new(123, 0)  
=> [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

Answer 2

Учитывая, что Ruby возвращает nil для несуществующего элемента (в отличие от ошибки типа index-of-of-bounds), вы можете просто использовать "или":

a = [1,2,3]
puts a[5]  # => nil
puts a[5] || "a default"  # => a default

YouМожно использовать метод «Обезьяньего патча», но вы, вероятно, не захотите делать это в чем-то большем, чем скрипт из одного файла:

a = [1,2,3]
def a.[](index)
  self.at(index) || "a default"
end
puts a[5]   # => "a default"

Answer 3

Я предложу элегантное решение Johans: foo.compact.inject(:+)

Answer 4

Если вы имеете дело с целыми числами, вы можете позвонить to_i:

foo = []
foo[100]
#=> nil
foo[100].to_i
#=> 0
foo[100] = 3
foo[100]
#=> 3

UPD

О, я не читал всю тему:)

так что вы можете использовать это:

foo.inject{|a,b| a.to_i + b.to_i }

который, на самом деле, не самый умный

Answer 5

Другой подход - переопределить метод Array#[] и вернуть значение по умолчанию, если нет элементов

class Array         
  def [](index)
     self.at(index) ? self.at(index) : 0
  end
end

и

arr = [1,2,3]
puts arr[0]  # print 1
puts arr[5]  # print 0

Answer 6

Я думаю, что массив - это неправильная абстракция, если вы хотите автоматически расширять массив. Добавьте еще один уровень абстракции.

Редактировать (из нашего обсуждения): Важно то, что код для достижения вашей цели расположен в правильном месте (принцип единой ответственности), и это место равно , а не Ваш "код клиента", следовательно, необходимость в новом классе. Расширение существующего класса Array (через наследование / mixin), вероятно, лучше, чем инкапсуляция искомого поведения в совершенно новом классе.

Answer 7

Самый простой способ будет:

new_array = Array.new(size, default_value)

Например:

new_array = Array.new(5,"foo")