使用 group_by 后根据条件转换哈希值

Question

我有一个 pipes 数组，它具有以下属性：pipe_id grade 和 grade_confidence.

我想在数组中查找与具有相同 ID 的其他对象具有相同属性的对象。我一直在使用 group_by 和 transform_values 来查找只有一个等级的 ID - 效果很好（感谢中的答案）。但是，如果可能的话，我仍然希望在最终结果中保留 grade_confidence。

class Pipes
  attr_accessor :pipe_id, :grade, :grade_confidence
  def initialize(pipe_id, grade, grade_confidence)
    @pipe_id = pipe_id
    @grade = grade
    @grade_confidence = grade_confidence
  end
end

pipe1 = Pipes.new("10001", "60", "A")
pipe2 = Pipes.new("10001", "60", "A")
pipe3 = Pipes.new("10001", "3", "F")
pipe4 = Pipes.new("1005", "15", "A")
pipe5 = Pipes.new("1004", "40", "A")
pipe6 = Pipes.new("1004", "40", "B")

pipes = []
pipes.push(pipe1, pipe2, pipe3, pipe4, pipe5, pipe6)

# We now have our array of pipe objects.

non_dups = pipes.group_by(&:pipe_id).transform_values { |a| a.map(&:grade).uniq }.select { |k,v| v.size == 1 }

puts non_dups

# => {"1005"=>["15"], "1004"=>["40"]}

需要

以上是我想要的——因为“10001”有两个不同的 grade，它被忽略了，“1004”和“1005”每个 ID 的等级相同。但是我想要的是也保留 grade_confidence，或者也根据条件包含 grade_confidence。

例如如果 grade_confidence 是 == "B" 最终结果将是 # => {"1004"=>["40", "B"]}

或

如果 grade_confidence 是 == "A" 最终结果将是 # => {"1005"=>["15", "A"], "1004"=>["40", "A"]}

是否可以调整 transform_values 以允许这样做，或者我需要走另一条路吗？

谢谢

Answer 1

您需要更新它：

non_dups = pipes
  .group_by(&:pipe_id)
  .transform_values { |a| [a.map(&:grade).uniq, a.map(&:grade_confidence)]}
  .select { |_, (grades, _confidences)| grades.size == 1 }
  .transform_values {|grades, confindeces| [grades.first, confindeces.sort.first]}

使用 group_by 后根据条件转换哈希值

Transforming hash values based on condition after using group_by

ruby

hash

group-by

hashmap