如何从二维对象数组中获取原始类型数组？

2024-03-12 18:58:23

从二维对象数组获取原始类型数组

在 Java 中，当我们需要操作一个数据表时，可能会遇到一个问题：如何从一个二维对象数组中获取列数据并返回原始类型数组。对于这个问题，有两个主要挑战：

1. 性能影响： 将接口更改为返回 Double[] 可能会对性能产生轻微影响。原始类型数组（如 double[]）比对象数组（如 Double[]）更紧凑、占用内存更少。此外，原始类型数组上的操作通常比对象数组上的操作更快。

2. 类型转换： Java 不允许将 Object[] 直接转换为 double[]。可以将其转换为 Double[]，但接口规定将数据返回为 double[]。

解决方法

有几种方法可以解决这个问题：

1. 使用流和映射：

可以使用流和映射对二维数组中的每一行进行操作，并将其转换为 double：

double[] getDoubles(int columnIndex) {
    return Arrays.stream(data)
            .mapToDouble(row -> (Double) row[columnIndex])
            .toArray();
}

2. 使用循环和装箱：

可以使用循环和装箱将 Object[][] 数据表中的每一行转换为 double：

double[] getDoubles(int columnIndex) {
    double[] doubles = new double[data.length];
    for (int i = 0; i < data.length; i++) {
        doubles[i] = (Double) data[i][columnIndex];
    }
    return doubles;
}

3. 使用 Arrays.stream() 和 asDoubleStream()：

可以使用 Arrays.stream() 和 asDoubleStream() 将 Object[][] 数据表中的每一行转换为 double：

double[] getDoubles(int columnIndex) {
    return Arrays.stream(data)
            .flatMapToDouble(Arrays::stream)
            .asDoubleStream()
            .toArray();
}