All Downloads are FREE. Search and download functionalities are using the official Maven repository.

hex.genmodel.utils.ArrayUtils Maven / Gradle / Ivy

There is a newer version: 3.46.0.6
Show newest version
package hex.genmodel.utils;

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

/**
 * Copied (partially) from water.util.ArrayUtils
 */
public class ArrayUtils {
  public static double[] nanArray(int len) {
    double[] arr = new double[len];
    for (int i = 0; i < len; i++) {
      arr[i] = Double.NaN;
    }
    return arr;
  }
  
  public static double l2norm(double[] x) {
    return Math.sqrt(l2norm2(x));
  }
  public static double l2norm2(double [] x){
    return l2norm2(x, false);
  }

  public static double l2norm2(double [] x, boolean skipLast){
    int last = x.length - (skipLast? 1 : 0);
    double sum = 0;
    for (int i = 0; i < last; ++i)
      sum += x[i]*x[i];
    return sum;
  }

  public static double[] flat(double[][] arr) {
    if (arr == null) return null;
    if (arr.length == 0) return null;
    int tlen = 0;
    for (double[] t : arr) tlen += (t != null) ? t.length : 0;
    double[] result = Arrays.copyOf(arr[0], tlen);
    int j = arr[0].length;
    for (int i = 1; i < arr.length; i++) {
      if (arr[i] == null)
        continue;
      System.arraycopy(arr[i], 0, result, j, arr[i].length);
      j += arr[i].length;
    }
    return result;
  }
  
  public static double[] eleDiff(final double[] from) {
    int arryLen = from.length-1;
    double[] cumsumR = new double[arryLen];
    for (int index = 0; index < arryLen; index++) {
      cumsumR[index] = from[index+1]-from[index];
    }
    return cumsumR;
  }

  public static int[] subtract(final int[] from, int val ) {
    int arryLen = from.length;
    int[] cumsumR = new int[arryLen];
    for (int index = 0; index < arryLen; index++) {
      cumsumR[index] = from[index]-val;
    }
    return cumsumR;
  }

  public static int[] subtract(final int[] from, int[] val ) {
    int arryLen = from.length;
    int[] cumsumR = new int[arryLen];
    for (int index = 0; index < arryLen; index++) {
      cumsumR[index] = from[index]-val[index];
    }
    return cumsumR;
  }

  public static double[] mult(double[] nums, double n) {
    assert !Double.isInfinite(n) : "Trying to multiply " + Arrays.toString(nums) + " by  " + n; // Almost surely not what you want
    if (nums != null)
      for (int i=0; i from[result] ) {
        result = i;
        maxCount = 1;
      } else if( from[i] == from[result] ) {
        if( rand.nextInt(++maxCount) == 0 ) result = i;
      }
    }
    return result;
  }

  public static int maxIndex(double[] from) {
    int result = 0;
    for (int i = 1; i < from.length; ++i)
      if (from[i] > from[result]) result = i;
    return result;
  }

  /**
   * Sort an integer array of indices based on values
   * Updates indices in place, keeps values the same
   * @param idxs indices
   * @param values values
   */
  public static void sort(int[] idxs, double[] values) {
    sort(idxs, values, 500);
  }
  public static void sort(int[] idxs, final double[] values, int cutoff) {
    if (idxs.length < cutoff) {
      //hand-rolled insertion sort
      for (int i = 0; i < idxs.length; i++) {
        for (int j = i; j > 0 && values[idxs[j - 1]] > values[idxs[j]]; j--) {
          int tmp = idxs[j];
          idxs[j] = idxs[j - 1];
          idxs[j - 1] = tmp;
        }
      }
    } else {
      Integer[] d = new Integer[idxs.length];
      for (int i = 0; i < idxs.length; ++i) d[i] = idxs[i];
      Arrays.sort(d, new Comparator() {
        @Override
        public int compare(Integer x, Integer y) {
        return values[x] < values[y] ? -1 : (values[x] > values[y] ? 1 : 0);
        }
      });
      for (int i = 0; i < idxs.length; ++i) idxs[i] = d[i];
    }
  }

  /**
   * Sort an integer array of indices based on values
   * Updates indices in place, keeps values the same
   * @param idxs indices
   * @param values values
   */
  public static void sort(final int[] idxs, final float[] values, int fromIndex, int toIndex, boolean abs, int increasing) {
    sort(idxs, values, fromIndex, toIndex, abs, increasing, 500);
  }

  public static void sort(final int[] idxs, final float[] values, int fromIndex, int toIndex, final boolean abs, final int increasing, int cutoff) {
    assert toIndex > fromIndex: "toIndex must be > fromIndex";
    if ((toIndex - fromIndex) < cutoff) {
      //hand-rolled insertion sort
      for (int i = fromIndex; i < toIndex; i++) {
        // the long line means: Sorted part of the array will be compared as absolute values if necessary
        for (int j = i; j > fromIndex && (abs ? Math.abs(values[idxs[j - 1]]) : values[idxs[j - 1]])*increasing > (abs ? Math.abs(values[idxs[j]]) : values[idxs[j]])*increasing; j--) {
          int tmp = idxs[j];
          idxs[j] = idxs[j - 1];
          idxs[j - 1] = tmp;
        }
      }
    } else {
      Integer[] d = new Integer[idxs.length];
      for (int i = 0; i < idxs.length; ++i) d[i] = idxs[i];
      Arrays.sort(d, fromIndex, toIndex, new Comparator() {
        @Override
        public int compare(Integer x, Integer y) {
          return Float.compare((abs ? Math.abs(values[x]) : values[x]) * increasing, (abs ? Math.abs(values[y]) : values[y]) * increasing);
        }
      });
      for (int i = 0; i < idxs.length; ++i) idxs[i] = d[i];
    }
  }

  public static String[] append(String[] a, String... b) {
    if (a==null ) 
      return b;
    String[] tmp = Arrays.copyOf(a,a.length + b.length);
    System.arraycopy(b, 0, tmp, a.length, b.length);
    return tmp;
  }

  public static String[][] append(String[][] a, String[]... b) {
    if (a==null )
      return b;
    String[][] tmp = Arrays.copyOf(a,a.length + b.length);
    System.arraycopy(b, 0, tmp, a.length, b.length);
    return tmp;
  }

  public static int[] append(int[] a, int... b) {
    if (a==null )
      return b;
    int[] tmp = Arrays.copyOf(a,a.length + b.length);
    System.arraycopy(b, 0, tmp, a.length, b.length);
    return tmp;
  }

  public static double[] signum(double[] array) {
    double[] signArray = new double[array.length];
    for (int i = 0; i < array.length; i++) {
      if (array[i] > 0)
        signArray[i] = 1;
      else if (array[i] < 0)
        signArray[i] = -1;
      else
        signArray[i] = 0;
    }
    return signArray;
  }

  public static double[] difference(double[] array) {
    double[] difference = new double[array.length - 1];
    for (int i = 0; i < array.length - 1; i++) {
      difference[i] = array[i+1] - array[i];
    }
    return difference;
  }

  /***
   * Carry out multiplication of row array a and matrix b and store the result in result array.  However the transpose
   * of the matrix is given.
   * 
   * @param a
   * @param bT
   * @param result
   */
  public static void multArray(double[] a, double[][] bT, double[] result) {
    int resultDim = result.length;
    int vectorSize = a.length;
    Arrays.fill(result, 0.0);
    for (int index = 0; index < resultDim; index++) {
      for (int innerIndex = 0; innerIndex < vectorSize; innerIndex++) {
        result[index] += a[innerIndex]*bT[index][innerIndex];
      }
    }
  }

  /**
   * Provide array from start to end in steps of 1
   * @param start beginning value (inclusive)
   * @param end   ending value (inclusive)
   * @return specified range of integers
   */
  public static int[] range(int start, int end) {
    int[] r = new int[end-start+1];
    for(int i=0;i




© 2015 - 2024 Weber Informatics LLC | Privacy Policy