【问题标题】:Groovy List: Group by by a list of elementsGroovy List:按元素列表分组
【发布时间】:2019-11-27 12:29:10
【问题描述】:

我基本上想对地图进行分组,我必须按汽车和颜色分组并将其价格相加,如果任何组价格大于25,则按汽车,颜色和电机将结果分组并乘以总和* 3,首先部分已完成,但我被困在第二组:

def row1 = ["car":'A',"color":'B',"motor":'C', "price": 12]
def row2 = ["car":'A',"color":'B',"motor":'C', "price": 12]
def row3 = ["car":'A',"color":'B',"motor":'D', "price": 2]
def row4 = ["car":'B',"color":'B',"motor":'D', "price": 13]
def arrayRows = []
arrayRows.add(row1)
arrayRows.add(row2)
arrayRows.add(row3)
arrayRows.add(row4)
println(arrayRows) //[[car:A, color:B, motor:C, price:12], [car:A, color:B, motor:C, price:12], [car:A, color:B, motor:D, price:2], [car:B, color:B, motor:D, price:13]]
def groups = arrayRows.groupBy {row -> [Car:row.car, Color:row.color] }.collect{k , v ->
    [
        car:k.Car,
        color:k.Color,
        motor:v.motor,
        price:v.collect { it.price }.sum()
    ]
}
println(groups) //[[car:A, color:B, motor:[C, C, D], price:26], [car:B, color:B, motor:[D], price:13]]
for(group in groups){
    if (group.price > 25){ //26>25
        println group //[car:A, color:B, motor:[C, C, D], price:26]
        //def groupByCarColorMotor = arrayRows.groupBy {[Car: group.car, Color: group.color, Motor:?]} | Must group by car, color and motor, and multiply * 3 it's price but since motor is an array, I'm not sure how to do so, I've tried groupBy { row -> group.it.motor} etc
    }
}

如果我有以下情况,我应该如何进行第二次分组: [车:A,颜色:B,电机:[C,C,D] 我应该按以下方式分组: [车:A,颜色:B,马达:C] 和 [汽车:A,颜色:B,电机:D]

预期的输出应该是: [ ["car":'A',"color":'B',"motor":'C', "price": 72, ["car":'A',"color":'B',"电机”:'D',“价格”:6]]

编辑; 我几乎设法做到了,问题是我得到了一组地图数组,而且你可能也会明白它背后的想法。

def arrayRows = [ 
        ["car":'A',"color":'B',"motor":'C', "price": 12],
        ["car":'A',"color":'B',"motor":'C', "price": 12],
        ["car":'A',"color":'B',"motor":'D', "price": 2],
        ["car":'B',"color":'B',"motor":'D', "price": 13]
        ]

println(arrayRows) //[[car:A, color:B, motor:C, price:12], [car:A, color:B, motor:C, price:12], [car:A, color:B, motor:D, price:2], [car:B, color:B, motor:D, price:13]]
def groups = arrayRows.groupBy {row -> [Car:row.car, Color:row.color] }.collect{k , v ->
    [
            car:k.Car,
            color:k.Color,
            price:v.collect { it.price }.sum()
    ]
}.findAll{it.price > 25}

def groupByCarColor = []
for (group in groups){
    groupByCarColor.add(arrayRows.findAll{ row -> row.car == group.car && row.color == group.color}.groupBy {row -> [Car:group.car, Color:group.color, Motor:row.motor] }.collect{k , v ->
        [
                car:k.Car,
                color:k.Color,
                motor:k.Motor,
                price:v.collect { it.price }.sum()*3
        ]
    })
}

输出:[[[汽车:A,颜色:B,电机:C,价格:72],[汽车:A,颜色:B,电机:D,价格:6]]]

【问题讨论】:

  • 价格总和不超过 25 的组的结果是什么?
  • @cfrick 我们不应该在第二组部分考虑这些。如果没有,输出应该是一个空数组。这会发生在 ["car":'B',"color":'B',"motor":'D', "price": 13] 因为所有汽车的价格和颜色(B,B ) 总和不要超过 25。

标签: dictionary groovy collections


【解决方案1】:

您的目标是通过三元组查找组 - 但您想过滤掉那些谓词在子组上失败的组。例如

def arrayRows = [ 
    [car:'A', motor:'C', color:'B',price: 12],
    [car:'A', motor:'C', color:'B',price: 12],
    [car:'A', motor:'D', color:'B',price:  2],
    [car:'B', motor:'D', color:'B',price: 13],
]


def output = arrayRows.inject([:].withDefault{ 0 }){ acc, row -> // group by the triplet and sum up the price
    acc[row.subMap(['car','motor','color'])]+=row.price; acc
}.groupBy{ k, _ -> // group by the filter criteria tuple
    k.subMap(['car','color'])
}.findAll{ _, v -> // eliminate the tuple-groups where price is to low
    v.values().sum() > 25
}.collectMany{ _, vs -> // reshape the data
    vs.collect{ k, v ->
        k + [price: v * 3]
    }
}

assert output==[[car:"A", motor:"C", color:"B", price:72], [car:"A", motor:"D", color:"B", price:6]]

【讨论】:

  • 基本上就是这样,cmets 也非常有帮助,命名我想要实现的目标也很有帮助!谢谢!我有另一个问题,我不确定是否应该编辑这个问题或创建另一个问题,如果我在地图中有另一个参数,我该怎么做,例如:[car:'A', motor:'C',颜色:'B',价格:15,参数:“hi”],[汽车:'A',电机:'C',颜色:'B',价格:11,参数:“hi”]。最终结果将是: [car:'A', motor:'C', color:'B',price: 26, parameter: ["hi", "hi"]] 在我看来,第一次注入是替换 groupBy 以避免问题解释
  • 您可以只用一个 groupBy 替换第一步(这是一个 groupBy 和总和),然后从那里开始工作。但是另一个参数提出了如何在分组步骤中聚合它的问题。如果它只是一个常数,您可以在上述管道的第一步中添加它(添加价格时)
【解决方案2】:

为你的家庭作业做一些直截了当的事情:

def arrayRows = [
["car":'A',"color":'B',"motor":'C', "price": 13],
["car":'A',"color":'B',"motor":'D', "price": 14],
["car":'B',"color":'B',"motor":'D', "price": 13],
]

List out = arrayRows.groupBy{ [ car:it.car, color:it.color ] }.inject( [] ){ List res, k, v ->
  if( 25 < v*.price.sum() ) v.each{ res << ( it + [ price:it.price * 3 ] ) }
  res
}

assert out.toString() == '[[car:A, color:B, motor:C, price:39], [car:A, color:B, motor:D, price:42]]'

【讨论】:

  • 虽然输出是正确的,但想象以下情况:["car":'A',"color":'B',"motor":'C', "price": 13] , ["car":'A',"color":'B',"motor":'C', "price": 14] 输出应该是:[[car:A, color:B, motor:C, price:81]] (13+14*3) 所以我要按 Car 和 Color 分组,条件为 25,然后再按第一个 groupBy 结果的 Car、Color 和 Motor 分组。不确定这是否是唯一的答案。我改变了例子
猜你喜欢
  • 1970-01-01
  • 2012-05-01
  • 1970-01-01
  • 2022-10-17
  • 1970-01-01
  • 1970-01-01
  • 2013-06-06
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多