如何從數據庫結果將行轉換為列

Question

我有一個由Postgres支持的Ruby項目。 我有一個表，該表具有一個名為first_ship_date的日期字段，以及一個交付狀態，即Rails枚舉。

我需要創建一個類似於以下內容的哈希或數組：

[ { week: 2016-01-04,
    0: 15,
    1: 30,
    2: 8
  },
  { week: 2016-01-11,
    0: 8,
    1: 45,
    2: 37
  }
]

完成此操作后，我將在Google圖表中顯示結果。

這是我到目前為止的內容：

SalesOrder.order("date_trunc('week', first_ship_date) ASC")
  .where("delivery_status is not null")
  .where("date(first_ship_date) >= ? AND date(first_ship_date) <=?", Date.new(2016,1,1), Date.new(2016,3,31))
  .pluck("date_trunc('week', first_ship_date)", :delivery_status)

這將返回一個數組，其值類似於：

[ [2016-01-04 00:00:00 UTC, 0],
  [2016-01-04 00:00:00 UTC, 2],
  [2016-01-04 00:00:00 UTC, 2],
  [2016-01-04 00:00:00 UTC, 2],
  [2016-01-04 00:00:00 UTC, 2] ]

將原始數據轉換為上面的哈希/數組的最佳和最有效的方法是什么，請記住枚舉的值並不總是可見的？ 意思是，有幾個星期我們沒有所有的delivery_status值。

Answer 1

回答

使用Enumerable#inject ： http : //ruby-doc.org/core-2.3.1/Enumerable.html#method-i-inject

一內膽：

your_array_or_arobject.inject({}) {|h,y| h[week = y[0].to_date.to_s] ||= []; h[week] << y[1]; h}.inject({}) {|h,(k,v)| h[k] = {week: k}.merge(v.each_with_index.inject({}) {|h,(k,i)| h[i] = k; h}); h}.values
#=> [{:week=>"2016-01-04", 0=>15, 1=>30, 2=>8}, {:week=>"2016-01-11", 0=>8, 1=>45, 2=>37}]

多行：

hash = your_array_or_arobject.inject({}) do |h,y| 
  h[week = y[0].to_date.to_s] ||= []
  h[week] << y[1]
  h
end

hash2 = hash.inject({}) do |h,(k,v)| 
  h[k] = {week: k}.merge(
    v.each_with_index.inject({}) do |h2,(k,i)| 
      h2[i] = k
      h2
    end
  )
  h
end

hash2.values
#=> [{:week=>"2016-01-04", 0=>15, 1=>30, 2=>8}, {:week=>"2016-01-11", 0=>8, 1=>45, 2=>37}]

---

證明

通過控制台：

irb(main):159:0> x = [
irb(main):160:1*   ["2016-01-04 00:00:00 UTC", 15],
irb(main):161:1*   ["2016-01-04 00:00:00 UTC", 30],
irb(main):162:1*   ["2016-01-04 00:00:00 UTC", 8],
irb(main):163:1* 
irb(main):164:1*   ["2016-01-11 00:00:00 UTC", 8],
irb(main):165:1*   ["2016-01-11 00:00:00 UTC", 45],
irb(main):166:1*   ["2016-01-11 00:00:00 UTC", 37]
irb(main):167:1> ]
=> [["2016-01-04 00:00:00 UTC", 15], ["2016-01-04 00:00:00 UTC", 30], ["2016-01-04 00:00:00 UTC", 8], ["2016-01-11 00:00:00 UTC", 8], ["2016-01-11 00:00:00 UTC", 45], ["2016-01-11 00:00:00 UTC", 37]]
irb(main):168:0> x.map! {|y| [DateTime.parse(y[0]), y[1]]}
=> [[Mon, 04 Jan 2016 00:00:00 +0000, 15], [Mon, 04 Jan 2016 00:00:00 +0000, 30], [Mon, 04 Jan 2016 00:00:00 +0000, 8], [Mon, 11 Jan 2016 00:00:00 +0000, 8], [Mon, 11 Jan 2016 00:00:00 +0000, 45], [Mon, 11 Jan 2016 00:00:00 +0000, 37]]
irb(main):169:0> hash = x.inject({}) {|h,y| h[week = y[0].to_date.to_s] ||= []; h[week]  {"2016-01-04"=>[15, 30, 8], "2016-01-11"=>[8, 45, 37]}
irb(main):170:0> hash.inject({}) {|h,(k,v)| h[k] = {week: k}.merge(v.each_with_index.inject({}) {|h,(k,i)| h[i] = k; h}); h}.values
=> [{:week=>"2016-01-04", 0=>15, 1=>30, 2=>8}, {:week=>"2016-01-11", 0=>8, 1=>45, 2=>37}]

---

UPDATE

重構每個值的出現次數。

一內膽：

your_array_or_arobject.inject({}) {|h,y| h[week = y[0].to_date.to_s] ||= []; h[week] << y[1]; h}.inject({}) {|h1,(k,v)| h1[k] = {week: k}.merge(v.inject({}) {|h2,k| h2[k] ||= 0; h2[k] += 1; h2}); h1}.values

多行：

hash = your_array_or_arobject.inject({}) do |h,y|
  h[week = y[0].to_date.to_s] ||= []
  h[week] << y[1]
  h
end

hash2 = hash.inject({}) do |h1,(k,v)| 
  h1[k] = {week: k}.merge(
    v.inject({}) do |h2,k| 
      h2[k] ||= 0
      h2[k] += 1
      h2
    end
  ) 
  h1
end

hash2.values

---

證明

通過控制台：

irb(main):001:0> require 'date'
=> true
irb(main):002:0> x = [
irb(main):003:1*   ['2016-01-04 00:00:00 UTC', 0], 
irb(main):004:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):005:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):006:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):007:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):008:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):009:1*   ['2016-01-04 00:00:00 UTC', 0],
irb(main):010:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):011:1*   ['2016-01-04 00:00:00 UTC', 0],
irb(main):012:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):013:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):014:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):015:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):016:1*   ['2016-01-04 00:00:00 UTC', 2],
irb(main):017:1*   ['2016-01-04 00:00:00 UTC', 2]
irb(main):018:1> ]
=> [["2016-01-04 00:00:00 UTC", 0], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 0], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 0], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2], ["2016-01-04 00:00:00 UTC", 2]]
irb(main):019:0> x.map! {|y| [DateTime.parse(y[0]), y[1]]}
=> [[#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 0], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 0], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 0], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2], [#<DateTime: 2016-01-04T00:00:00+00:00 ((2457392j,0s,0n),+0s,2299161j)>, 2]]
irb(main):020:0> x.inject({}) {|h,y| h[week = y[0].to_date.to_s] ||= []; h[week] << y[1]; h}.inject({}) {|h1,(k,v)| h1[k] = {week: k}.merge(v.inject({}) {|h2,k| h2[k] ||= 0; h2[k] += 1; h2}); h1}.values
=> [{:week=>"2016-01-04", 0=>3, 2=>12}]