如何在Ruby中設置固定數量的數組元素

Question

如何在Ruby中設置數組的固定數量的元素。

例如。 a=["a","b","c","d"] 設置數組大小設置爲3將輸出

a=["a","b","cd"] 

Answer 1

class Array 
    def squeeze(n, &p) 
    p = Proc.new {|xs| xs.join} unless p 
    arr = self[0..n-2] 
    arr << p.call(self[n-1..-1]) 
    end 
end 

a = ['a', 'b', 'c', 'd', 'e'] 
a.squeeze(3) # => ["a", "b", "cde"] 

它需要邊界檢查，但你的想法。請注意，「合併」功能可以作爲一個塊參數：

[1, 2, 3, 4].squeeze(3) {|xs| xs.inject {|acc,x| acc+x}} # => [1, 2, 7] 

Answer 2

如果你知道的元素只是一個個字符的字符串，你可以這樣做：

a.join.split '', 3 

否則：

a[0..1] + [a[2..-1].join] 

或許：

a[0..1] << a[2..-1].join