Question

我想在数组或arraylist或Java中的其他数据结构上使用libmylib.a函数，但我没有在文档中找到argsort函数。我知道argsort类上有一个sort方法，但没有任何Collections方法。是否已经实施了argsort方法的课程？

Answer 1

我找到了这门课，我希望这是你想要的：

import java.util.Arrays;
import java.util.Comparator;
import java.util.Random;

public final class ArrayUtils {
    public static int[] argsort(final float[] a) {
        return argsort(a, true);
    }

    public static int[] argsort(final float[] a, final boolean ascending) {
        Integer[] indexes = new Integer[a.length];
        for (int i = 0; i < indexes.length; i++) {
            indexes[i] = i;
        }
        Arrays.sort(indexes, new Comparator<Integer>() {
            @Override
            public int compare(final Integer i1, final Integer i2) {
                return (ascending ? 1 : -1) * Float.compare(a[i1], a[i2]);
            }
        });
        return asArray(indexes);
    }

    public static <T extends Number> int[] asArray(final T... a) {
        int[] b = new int[a.length];
        for (int i = 0; i < b.length; i++) {
            b[i] = a[i].intValue();
        }
        return b;
    }

    public static double[] castOf(final float[] x) {
        double[] y = new double[x.length];
        for (int i = 0; i < y.length; i++) {
            y[i] = x[i];
        }
        return y;
    }

    public static int[] castOf(final long[] original) {
        return castOf(original, original.length);
    }

    public static int[] castOf(final long[] original, final int newLength) {
        int[] cast = new int[newLength];
        int length = Math.min(cast.length, original.length);
        for (int i = 0; i < length; i++) {
            long o = original[i];
            if (o < Integer.MIN_VALUE || o > Integer.MAX_VALUE) {
                throw new IllegalArgumentException();
            }
            cast[i] = (int) o;
        }
        return cast;
    }

    public static float[][] copyOf(final float[][] x, final int newLength) {
        float[][] y = new float[newLength][];
        for (int i = 0; i < y.length; i++) {
            if (x[i] != null) {
                y[i] = Arrays.copyOf(x[i], x[i].length);
            }
        }
        return y;
    }

    /**
     * Assigns a random value to each element of the specified array of doubles.
     */
    public static void fillRandom(final double[] x, final Random rng) {
        for (int i = 0; i < x.length; i++) {
            x[i] = rng.nextDouble();
        }
    }

    private ArrayUtils() {
    }
}

链接： Code

Answer 2

截至2017年，您可以找到argsort function in the ND4J library（可能是最接近numpy的Java）。它没有进入2017年的0.9.1版本，但它看起来好像将在即将发布的0.9.2版本中。代码实际上是从JetStream复制到其答案中的array4j代码派生的。 array4j现在似乎已被抛弃，因此只要argsort成为一个版本，ND4J就可能更好了。 0.9.2一旦发布，您就可以从Maven Central repository安装它。

顺便说一下，你在评论JetStream的回答中提到了

我想知道为什么没有这个argsort方法的类。它存在于Python numpy中，例如......

这不是一个公平的比较，因为numpy不是Python标准库的一部分，不仅仅是ND4J用于Java。这两种语言都不包含标准库中的argsort，并且都可以在第三方库中使用。

java中是否有argsort函数？

2 个答案: