zzh
/
gatk-3.8
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added ability to sample from a list *without* replacement. 

git-svn-id: file:///humgen/gsa-scr1/gsa-engineering/svn_contents/trunk@1304 348d0f76-0448-11de-a6fe-93d51630548a
This commit is contained in:
kiran 2009-07-23 21:00:19 +00:00
parent 038cbcf80e
commit 7c20be157c
1 changed files with 36 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
java/src/org/broadinstitute/sting/utils/ListUtils.java
View File

@ -1,9 +1,6 @@
package org.broadinstitute.sting.utils; package org.broadinstitute.sting.utils;
import java.util.List; import java.util.*;
import java.util.HashSet;
import java.util.ArrayList;
import java.util.Random;
/** /**
* Created by IntelliJ IDEA. * Created by IntelliJ IDEA.
@ -16,8 +13,14 @@ public class ListUtils {
static Random rand = new Random(12321); //System.currentTimeMillis()); static Random rand = new Random(12321); //System.currentTimeMillis());
/**
* Returns n random indices drawn with replacement from the range 0..(k-1)
*
* @param n the number of random indices to draw (with replacement)
* @param k the total number of indices allowed
* @return a list of random indices ranging from 0 to (k-1) with possible duplicates
*/
static public ArrayList<Integer> sampleIndicesWithReplacement(int n, int k) { static public ArrayList<Integer> sampleIndicesWithReplacement(int n, int k) {
// Returns n random indices drawn with replacement from the range 1..k
ArrayList<Integer> chosen_balls = new ArrayList <Integer>(); ArrayList<Integer> chosen_balls = new ArrayList <Integer>();
for (int i=0; i<n; i++) { for (int i=0; i<n; i++) {
@ -29,10 +32,35 @@ public class ListUtils {
return chosen_balls; return chosen_balls;
} }
static public <T> ArrayList<T> sliceListByIndices(List<Integer> indices, List<T> list) { /**
// Given a list of indices into a list, return those elements of the list with the possibility * Returns n random indices drawn without replacement from the range 0..(k-1)
// of drawing list elements multiple times *
* @param n the number of random indices to draw (without replacement)
* @param k the total number of indices allowed
* @return a list of random indices ranging from 0 to (k-1) without duplicates
*/
static public ArrayList<Integer> sampleIndicesWithoutReplacement(int n, int k) {
ArrayList<Integer> chosen_balls = new ArrayList<Integer>(k);
for (int i = 0; i < n; i++) {
chosen_balls.add(i);
}
Collections.shuffle(chosen_balls, rand);
//return (ArrayList<Integer>) chosen_balls.subList(0, k);
return new ArrayList<Integer>(chosen_balls.subList(0, k));
}
/**
* Given a list of indices into a list, return those elements of the list with the possibility of drawing list elements multiple times
* @param indices the list of indices for elements to extract
* @param list the list from which the elements should be extracted
* @param <T> the template type of the ArrayList
* @return a new ArrayList consisting of the elements at the specified indices
*/
static public <T> ArrayList<T> sliceListByIndices(List<Integer> indices, List<T> list) {
ArrayList<T> subset = new ArrayList<T>(); ArrayList<T> subset = new ArrayList<T>();
for (int i : indices) { for (int i : indices) {