配列内の配列の合計の求め方

Question

次のような配列があるときに、配列内の配列の１つ目が”red”、２つ目が”apple”の場合の配列内の配列の３つ目の合計はどうやって書けば良いのでしょうか？

list = [["red","apple",50],["blue","grape",110],["red","remon"20],["red","apple",100],["yellow","banana",30]]

-> 50+100で150と出力させたい。

下記の様にやってみましたが上手くいきませんでした・・

puts list.select {|x| x[0] == "red" && x[1] == "apple"}.select{|x| x[2].to_f}.sum(&:last)

Answer 1

それで、150と出ますよ。
2つ目のselectは無意味なので、

puts list.select {|x| x[0] == "red" && x[1] == "apple"}.sum(&:last)

でいいです。整数じゃなくてFloatで計算したかったのなら、selectじゃなくてmapです。

puts list.select {|x| x[0] == "red" && x[1] == "apple"}.map{|x| x[2].to_f}.sum

です。

Comments

[kzd847686]

ありがとうございます！

Answer 2

以前の最大値を求めたときのものを改良したものです。
selectでもいいのではって気もしないでもないですけど...

list.map{|x, y, z| z.to_f if x=='red' && y=='apple'}.compact.sum

下記の様にやってみましたが上手くいきませんでした・・

list = [["red","apple",50],["blue","grape",110],["red","remon"20],["red","apple",100],["yellow","banana",30]]

3つ目の要素の"remon"と20の間に,がないから...とかではないですか？

Comments

[kzd847686]

ありがとうございます！

Answer 3

少しずつ違う複数の方法で書いてみました。

list = [
  ["red", "apple", 50],
  ["blue", "grape", 110],
  ["red", "remon", 20],
  ["red", "apple", 100],
  ["yellow", "banana", 30]
]

sum = list.map {|x| (x[0] == "red" && x[1] == "apple") ? x[2].to_f : 0 }.sum
p sum

sum = list.select {|x| x[0] == "red" && x[1] == "apple"}.map{|x| x[2].to_f}.sum
p sum


sum = list.select {|x| x[0] == "red" && x[1] == "apple"}.map(&:last).sum(&:to_f)
p sum

g = list.group_by {|x| x[0..1]}.each_with_object({}) {|(k, v), mem| mem[k] = v.map(&:last).sum(&:to_f)}
p g
p g[["red", "apple"]]

g = list.group_by {|x| x[0..1]}.each_with_object({}) {|(k, v), mem| mem[k] = v.map(&:last).map(&:to_f)}
p g
p g[["red", "apple"]].sum

実行例

Comments

[kzd847686]

ありがとうございます！いくつかの方法で教えて頂き勉強になりました。

Answer 4

言語を間違いました。