3

我有以下列表:

[FOO,1]
[FOO,2]
[BAR,1]
[BAR,2]
[BAR,3]

我想把它转换成这样的地图:

FOO:[1,2]
BAR:[1,2,3]

我觉得答案可能与使用 groupBy 或注入有关,但我还不够“时髦”,无法直观地知道最好的方法。想法?

4

2 回答 2

7

所以,给定:

def list = [ ['FOO',1],
             ['FOO',2],
             ['BAR',1],
             ['BAR',2],
             ['BAR',3] ]

Yu 可以按每个列表中的第一个元素进行分组

def map = list.groupBy { it[ 0 ] }

这给了你:

[ 'FOO': [ ['FOO', 1], ['FOO', 2] ],
  'BAR': [ ['BAR', 1], ['BAR', 2], ['BAR', 3] ] ]

因此,您需要将值列表中的每个条目更改为列表中的第二个元素:

def result = list.groupBy { it[ 0 ] }
                 .collectEntries { key, value ->
                     [ key, value*.getAt( 1 ) ]
                 }

这给了你:

[ 'FOO': [ 1, 2 ],
  'BAR': [ 1, 2, 3 ] ]

按照要求 :-)

于 2013-10-24T14:27:48.000 回答
7

虽然 tim_yates 的回答是正确的,但我想指出一个替代解决方案:

def result = [:].withDefault { [] } // initialize new keys with empty list
list.each {
  result[it[0]] << it[1]
}

这种方法只使用一个循环,执行时间应该与列表大小成线性关系。

于 2013-10-24T17:00:23.150 回答