Как расширить / уменьшить двумерный Ruby NArray?

Question

Если у меня есть Narray с формой 100, 10000 и я хочу расширить его до 100, 20000 (в основном, добавить строки), каков правильный способ достижения этого? Для расширения массивных Narrays я хотел бы избежать использования временного Narrays по соображениям памяти.

Answer 1

require "narray"

class NArray
  def expand(*new_shape)
    na = NArray.new(self.typecode,*new_shape)
    range = self.shape.map{|n| 0...n}
    na[*range] = self
    return na
  end
end

p a = NArray.float(2,3).indgen!
# => NArray.float(2,3):
#    [ [ 0.0, 1.0 ],
#      [ 2.0, 3.0 ],
#      [ 4.0, 5.0 ] ]

p a.expand(3,4)
# => NArray.float(3,4):
#    [ [ 0.0, 1.0, 0.0 ],
#      [ 2.0, 3.0, 0.0 ],
#      [ 4.0, 5.0, 0.0 ],
#      [ 0.0, 0.0, 0.0 ] ]

Нет общего способа расширить блок памяти без движения. Блок памяти может быть увеличен только в том случае, если за ним достаточно свободного места, но такой случай обычно бывает неожиданным.