如何从 Java 8 中的流创建二维数组？

Question

我有一个这样的文本文件：

ids.txt

我想读取此文件并将其加载到二维数组中。我希望有一个类似于下面的数组：

Object[][] data = new Object[][] { //
     { new Integer(1000) }, //
     { new Integer(999) }, //
     { new Integer(745) }, //
     { new Integer(123) }, //
     ...
};

这是我写的代码：

File idsFile = ... ;
try (Stream<String> idsStream = Files.lines(idsFile.toPath(), StandardCharsets.US_ASCII)) {
    Object[][] ids = idsStream
       .filter(s -> s.trim().length() > 0)
       .toArray(size -> new Object[size][]);

    // Process ids array here...
}

当运行这段代码时，抛出异常：

java.lang.ArrayStoreException: null
at java.lang.System.arraycopy(Native Method) ~[na:1.8.0_45]
at java.util.stream.SpinedBuffer.copyInto(Unknown Source) ~[na:1.8.0_45]
at java.util.stream.Nodes$SpinedNodeBuilder.copyInto(Unknown Source) ~[na:1.8.0_45]
at java.util.stream.SpinedBuffer.asArray(Unknown Source) ~[na:1.8.0_45]
at java.util.stream.Nodes$SpinedNodeBuilder.asArray(Unknown Source) ~[na:1.8.0_45]
at java.util.stream.ReferencePipeline.toArray(Unknown Source) ~[na:1.8.0_45]
...

如何解决这个异常？

Answer 1

您的最后一行可能应该是 size -> new Object[size]，但是您需要提供大小为 1 的整数数组，并且您还需要将字符串解析为整数。

我建议如下：

try (Stream<String> idsStream = Files.lines(idsFile.toPath(), StandardCharsets.US_ASCII)) {
    Object[][] ids = idsStream
       .map(String::trim)
       .filter(s -> !s.isEmpty())
       .map(Integer::valueOf)
       .map(i -> new Integer[] { i })
       .toArray(Object[][]::new);

    // Process ids array here...
}

Answer 2

给定一个 Stream<String>，您可以将每个项目解析为一个 int，然后使用以下方法将其包装成一个 Object[]：

 strings
        .filter(s -> s.trim().length() > 0)
        .map(Integer::parseInt)
        .map(i -> new Object[]{i})

现在要将该结果转换为 Object[][]，您只需执行以下操作：

Object[][] result = strings
        .filter(s -> s.trim().length() > 0)
        .map(Integer::parseInt)
        .map(i -> new Object[]{i})
        .toArray(Object[][]::new);

对于输入：

final Stream<String> strings = Stream.of("1000", "999", "745", "123");

输出：

[[1000], [999], [745], [123]]

如何从 Java 8 中的流创建二维数组？

How to create a two dimensional array from a stream in Java 8?

java

java-8

java-stream