如何将Java 8 Stream转换为二维数组？

Question

我正在尝试将基于地图的Stream转换为二维数组。 我已经想出如何将它存储在一维数组中。 这是工作代码片段：

Float[] floatArray = map.entrySet()
                        .stream()
                        .map(key -> key.getKey().getPrice())
                        .toArray(size -> new Float[size]);

当我执行上面的代码时，我按预期填充了我的Float数组。 现在我需要将它扩展到一个二维数组，我需要将结果存储在这些行的二维数组的第一维中：

Float[][1] floatArray = map.entrySet()
                           .stream()
                           .map(key -> key.getKey().getPrice())
                           .toArray(size -> new Float[size][1]);

上面的代码不起作用。 你能告诉我如何使用Java 8流完成这项任务吗？ 提前致谢！

Answer 1

如果您查看<A> A[] toArray(IntFunction<A[]> generator) ，您会看到它将Stream<A>转换为A[] ，这是A元素的一维数组。 因此，为了创建2D数组， Stream的元素必须是数组。

因此，如果首先将Stream的元素map到1D数组，然后调用toArray ，则可以创建2D数组：

Float[][] floatArray = 
    map.entrySet()
       .stream()
       .map(key -> new Float[]{key.getKey().getPrice()})
       .toArray(size -> new Float[size][1]);

Answer 2

您可以使用以下内容：

Float[][] array = map.entrySet()
    .stream()
    .map(Map.Entry::getKey)
    .map(YourKeyClass::getPrice) // 1)
    .map(price -> new Float[]{ price })
    .toArray(Float[][]::new);

这就像你期望的那样创建一个2D数组。

注意：通过注释1）你必须用包含方法getPrice()的类替换YourKeyClass ，返回一个Float对象。

---

另一种方法是使用.keySet()而不是.entrySet() ：

Float[][] array = map.keySet().stream()
    .map(YourKeyClass::getPrice)
    .map(price -> new Float[]{price})
    .toArray(Float[][]::new);