Как сгруппировать объекты из списка, который может принадлежать двум или более группам? - PullRequest
Новая
       13

Как сгруппировать объекты из списка, который может принадлежать двум или более группам?

6 голосов
/ 27 марта 2019

У меня есть список предметов, где каждый предмет может принадлежать к одной или нескольким категориям.Для ограниченного набора категорий (строка) я хочу создать карту с категорией в качестве ключа и списком элементов в качестве значения.

Предположим, что мой класс Item определен, как показано ниже: 

public static class Item{
    long id;
    List<String> belongsToCategories;

    public List<String> getBelongsToCategories() {
        return belongsToCategories;
    }

    public void setBelongsToCategories(List<String> belongsToCategories) {
        this.belongsToCategories = belongsToCategories;
    }

    public Item(long id,List<String> belongsToCategories) {
        this.id = id;
        this.belongsToCategories = belongsToCategories;
    } 

    @Override
    public String toString() {
        return "Item{" + "id=" + id + '}';
    }        
}

и список элементов: 

public static void main(String[] args) {
    List<Item> myItemList   = new ArrayList<>();

    myItemList.add(new Item(1,Arrays.asList("A","B")));
    myItemList.add(new Item(2,Arrays.asList("A","C")));
    myItemList.add(new Item(3,Arrays.asList("B","C")));
    myItemList.add(new Item(4,Arrays.asList("D")));
    myItemList.add(new Item(5,Arrays.asList("D","E")));
    myItemList.add(new Item(6,Arrays.asList("A","F")));

    Map<String,List<Item>> myMap= new HashMap<>();

Как я могу заполнить myMap из myList?

Я думал, что Stream API может помочь, но я не знаю, какой классификатор добавить в метод groupingBy, когда Item может принадлежать к одной или нескольким категориям 

 myItemList.stream().collect(Collectors.groupingBy(classifier));

This 

myItemList.stream().collect(Collectors.groupingBy(Item::getBelongsToCategories));

производит 

[D, E]=[Item{id=5}]
[B, C]=[Item{id=3}]
[A, B]=[Item{id=1}]
[D]=[Item{id=4}]
[A, C]=[Item{id=2}]
[A, F]=[Item{id=6}]

ожидается что-то вроде: 

A=[Item{id=1}, Item{id=2}, Item{id=6}]
B=[Item{id=1}, Item{id=3}]
C=[Item{id=2}, Item{id=3}]
D=[Item{id=4}, Item{id=5}]
E=[Item{id=5}]
F=[Item{id=6}]

Ответы [ 3 ]

3 голосов
/ 29 марта 2019

Вы можете использовать groupByEach из Коллекции Eclipse . 

Multimap<String, Item> itemsByCategory =
        ListIterate.groupByEach(myItemList, Item::getBelongsToCategories);

System.out.println(itemsByCategory);

Выход: 

{D=[Item{id=4}, Item{id=5}], 
E=[Item{id=5}], 
F=[Item{id=6}], 
A=[Item{id=1}, Item{id=2}, Item{id=6}], 
B=[Item{id=1}, Item{id=3}], 
C=[Item{id=2}, Item{id=3}]}

Вы также можете использовать служебный класс Collectors2 иgroupByEach Сборщик с Java Stream.

Multimap<String, Item> itemsByCategory = myItemList.stream().collect(
        Collectors2.groupByEach(
                Item::getBelongsToCategories,
                Multimaps.mutable.list::empty));

Примечание: я являюсь коммиттером для коллекций Eclipse.

2 голосов
/ 27 марта 2019

Вы можете использовать flatMap для сопоставления с SimpleEntry, а затем groupingBy как: 

return items.stream()
        .flatMap(p -> p.getBelongsToCategories()
                .stream()
                .map(l -> new AbstractMap.SimpleEntry<>(l, p)))
        .collect(Collectors.groupingBy(Map.Entry::getKey,
                Collectors.mapping(Map.Entry::getValue,
                        Collectors.toList())));
1 голос
/ 28 марта 2019

Вот решение ListMultimap, либо из Google Guava , либо из моей библиотеки Abacus-Util 

ListMultimap<String, Item> result = Multimaps.newListMultimap(new HashMap<>(), () -> new ArrayList<>()); // by Google Guava.
// Or result = N.newListMultimap(); // By Abacus-Util
items.forEach(item -> item.getBelongsToCategories().forEach(c -> result.put(c, item)));

Я думаю, что это будет немного более эффективно и сжато, без создания ненужных временных Entry объектов flatMap.

...