Сбор коллекции списков на основе аналогичного индекса

Question

У меня есть коллекция вроде: List<List<Object>> firstList

Я хочу сгруппировать похожий список шаблонов: List<List<Object>> secondList, но сгруппированы по индексам.

скажем

firstList [1]:
   0 = {Object A}"
   1 = {Object B}"
   2 = {Object C}"

firstList [2]:
   0 = {Object A}"
   1 = {Object B}"
   2 = {Object C}"

Я хочу сгруппировать эту коллекцию как

secondList [1]:
   0 = {Object A}"
   1 = {Object A}"

secondList [2]:
   0 = {Object B}"
   1 = {Object B}"

secondList [3]:
   0 = {Object C}"
   1 = {Object C}"

До сих пор я пробовал

for (int i = 0; i <firstList.size() ; i++) {
    List<Object> list = firstList.get(i);
    List<Object> rlPr = new ArrayList<>();

    for (int j = 0; j <list.size()-1; j++) {
        rlPr.add(list.get(i));
    }
    secondList.add(rlPr);
}

Но я не получаю ожидаемого.Я использую Java 8.

РЕДАКТИРОВАТЬ: ВЕСЬ СПИСОК ОДНОГО РАЗМЕРОВ

Answer 1

Потоковый способ сделать это (заменить Object на CustomObject):

List<List<CustomObject>> secondList = new ArrayList<>( // Collection to ArrayList
       firstList.stream() // Stream<List<CustomObject>>
                .flatMap(Collection::stream) // Stream<CustomObject>
                .collect(Collectors.groupingBy(Function.identity())) // Map<CustomObject, List<CustomObject>>
                .values()); // Collection<List<CustomObject>>

Answer 2

Вы можете использовать Map для группировки значений по индексу:

Map<Integer, List<Object>> map = new TreeMap<>();
for (List<Object> objects : firstList) {
    for (int i = 0, l = objects.size(); i < l; i++) {
        map.computeIfAbsent(i, k -> new ArrayList<>()).add(objects.get(i));
    }
}

А затем для возврата List:

List<List<Object>> secondList = new ArrayList<>(map.values());

Answer 3

Сначала получите максимальную длину подсписка.В вашем случае оба равны 3. Итерируйте по всем этим индексам, получая значения из каждого подсписка по этому индексу и собирая их в новый подсписок.

final int maxLengthOfSublist = list1.stream().mapToInt(List::size).max().orElse(0);
final List<List<Integer>> list2 = IntStream.range(0, maxLengthOfSublist)
    .mapToObj(idx -> list1.stream()
        .filter(sublist -> idx < sublist.size())
        .map(sublist -> sublist.get(idx))
        .collect(Collectors.toList())
    )
    .collect(Collectors.toList());

Это будет работать, даже если подсписки имеют разную длину.

Answer 4

Предполагая, что списки внутри основного списка имеют одинаковый размер, вы можете сделать что-то вроде этого ...

    import java.util.ArrayList;
    import java.util.Arrays;
    import java.util.Collections;
    import java.util.List;

    public class OhCrikey {

        public static void main(String[] args) {
            List<List<String>> lists = Arrays.asList(
                    Arrays.asList("1a", "2a", "3a"),
                    Arrays.asList("1b", "2b", "3b"),
                    Arrays.asList("1c", "2c", "3c")
            );

            List<List<String>> result = transform(lists);

            result.forEach(System.out::println);
        }

        private static <T> List<List<T>> transform(List<List<T>> lists) {
            if(lists == null || lists.isEmpty()) {
                return Collections.emptyList();
            }
            // Check each sub-list contains the same number of elements
            checkAllSameSize(lists);

            // Initialise the results list
            List<List<T>> result = new ArrayList<>();

            // Get the size of each sub-list
            int totalLists = lists.get(0).size();

            // Fill up the results list with 'totalLists' empty lists
            for(int k = 0; k < totalLists; k++) {
                result.add(new ArrayList<>());
            }

            // For each input list...
            lists.forEach(list -> {
                // Iterate over it and add the kth element to the kth result list.
                for(int k = 0; k < list.size(); k++) {
                    result.get(k).add(list.get(k));
                }
            });

            return result;
        }

        private static <T> void checkAllSameSize(List<List<T>> lists) {
            int size = lists.get(0).size();

            // Make sure each list has the same size as the first list
            lists.forEach(list -> {
                if(list.size() != size) {
                    throw new AssertionError("All child lists must have same number of elements.");
                }
            });
        }
    }

Печать ...

    [1a, 1b, 1c]
    [2a, 2b, 2c]
    [3a, 3b, 3c]