使用Java流获取嵌套列表的第N个元素

Question

下面是一个函数，它将获得一个嵌套的ArrayListAllLists、indexnth，并返回每个子列表的第n个元素。

例如。对于alllists={{1,2,3},{4,5,6},{7,8,9}},nth=1,函数将返回{2,5,8}

public static String[] getEveryNthElement(ArrayList<ArrayList<String>> allLists, int nth) {
        String[] nthList = new String[allLists.size()];
        int n = 0;
        for (ArrayList<String> sList: allLists) {
            if (nth <= sList.size()) {
                nthList[n] = (sList.get(nth));
            }
            n += 1;
        }
        return nthList;
    }

我已经找到了一个可以打印出来的版本：

group.stream()
    .forEach(items -> {
        System.out.println(items.get(1));  // prints 2, 5, 8
});

如何将结果汇总成清单？

Answer 1

已解决：

public static String[] getEveryNthElement(ArrayList<ArrayList<String>> allLists, int nth){
return allLists.stream()
               .map(list -> list.get(nth))
               .collect(Collectors.toList());
}