如何在Java 8中从流创建一个二维数组？

Question

I have a text file like this:

ids.txt

1000
999
745
123
...

我希望能够读取该文件并将其加载到二维数组中。我期望得到一个类似下面的数组：

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]
... 

如何解决这个异常？

- Stephan

---

为什么要使用 new Integer(1000)？而且你想要一个 [][]，其中内部数组始终具有 length == 1？ - Boris the Spider

@BoristheSpider 这些ID被传递给一个第三方库，该库仅接受Object [][ ]作为输入。 - Stephan

Answer 1

如果给你一个 Stream<String>，你可以将其解析为 int，并将其包装为 Object[] 使用以下代码：

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

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

现在，要将该结果转换为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]]

- Boris the Spider

---

如果你怀疑字符串可能会时不时地出现前后有空格的情况，那么最好添加单独的“trim”步骤：strings.map(String::trim).filter(s -> !s.isEmpty())。此外，我建议使用Integer::valueOf而不是Integer::parseInt，因为它更符合所需的类型（Integer而不是int）。虽然这可能只是个人喜好问题。 - Tagir Valeev

@TagirValeev 为什么“最好添加单独的修剪步骤”？ - Stephan

@Stephan：这样，您就可以解析那些数字后面带空格的文件（例如“1 \n2\n3 \n”）。当前，您可能会在这样的文件上遇到NumberFormatException异常。 - Tagir Valeev

你认为这两个 map 操作有益处，而不是简单地说 .map(s -> new Object[]{Integer.parseInt(s)})？ - Holger

@Holger 我不是很确定。我喜欢将每个逻辑转换拆分成单独的 map 步骤 - 我认为这使得代码更易读。我想这只是个人偏好的问题 - 我怀疑性能影响不大。 - Boris the Spider

Answer 2

你最后一行应该是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...
}

- assylias