Ruby: Cách đặt biến thành 0 hoặc, nếu nó đã được đặt, tăng thêm 1

Question

Tôi biết toán tử || =, nhưng đừng nghĩ rằng nó sẽ giúp tôi ở đây ... cố gắng tạo một mảng đếm số "loại" trong một mảng các đối tượng.

array.each do |c| 
    newarray[c.type] = newarray[c.type] ? newarray[c.type]+1 ? 0 
end 

Có cách nào duyên dáng hơn để thực hiện việc này không?

Answer 1

types = Hash.new(-1) # It feels like this should be 0, but to be 
        # equivalent to your example it needs to be -1 
array.each do |c| 
    types[c.type] += 1 
end 

Answer 2

||= không giúp đỡ:

types = {} 
array.each do |c| 
    types[c.class] ||= 0 
    types[c.class] += 1 
end 

Answer 3

array.each do |c| 
    newarray[c.type] = 1 + (newarray[c.type] || -1) 
end 

Ngoài ra

array.each do |c| 
    newarray[c.type] ||= -1 
    newarray[c.type] += 1 
end 

Answer 4

Sử dụng Array#fetch phương pháp mà bạn có thể cung cấp một giá trị mặc định nếu chỉ số không tồn tại:

array.each do |c| 
    newarray[c.type] = newarray.fetch(c.type, -1) + 1 
end 

Answer 5

Biến số newarray của bạn được đặt tên kỳ quặc, vì trong Ruby và hầu hết các ngôn ngữ khác, mảng được lập chỉ mục theo số nguyên, không phải các đối tượng ngẫu nhiên như Class. Có nhiều khả năng đây là Hash.

Ngoài ra, bạn nên sử dụng c.class, thay vì c.type, không được dùng nữa.

Cuối cùng, vì bạn đang tạo một Hash, bạn có thể sử dụng inject như vậy:

newarray = array.inject({}) do |h,c| 
    h[c.class] = h.key?(c.class) ? h[c.class]+1 : 0 
    h 
end 

Hoặc, đối với một lớp lót:

newarray = array.inject({}) { |h,c| h[c.class] = h.key?(c.class) ? h[c.class]+1 : 0 ; h } 

Như bạn có thể thấy, điều này mang lại kết quả mong muốn:

irb(main):001:0> array = [1, {}, 42, [], Object.new(), [1, 2, 3]] 
=> [1, {}, 42, [], #<Object:0x287030>, [1, 2, 3]] 
irb(main):002:0> newarray = array.inject({}) { |h,c| h[c.class] = h.key?(c.class) ? h[c.class]+1 : 0 ; h } 
=> {Object=>0, Hash=>0, Array=>1, Fixnum=>1} 

Answer 6

Trong Ruby 1.8.7 hoặc mới hơn, bạn có thể sử dụng group_by và sau đó chuyển từng danh sách các phần tử thành số đếm - 1 và tạo một băm từ mảng được trả về bởi map.

Hash[array.group_by(&:class).map { |k,v| [k, v.size-1] }]