天天看點

java中位數_java 計算中位數方法

最近工作需要 要求把python的代碼寫成java版本,python中有一個np.median()求中位數的方法,java決定手寫一個

先說說什麼是中位數:

中位數就是中間的那個數,

如果一個集合是奇數個,那麼中位數就是按大小排列後,最中間那個數,

如果一個集合是偶數個,那麼中位數就是按大小排列後,最中間那2個數的平均數。

比如:

1,2,3,4,5  那中位數就是3

1,2,3,4,5,6 那中位數就是 (3+4)/2 = 3.5

知道邏輯後方法就很簡單了 下面是代碼

public static void main(String[] args) {

List total = new ArrayList();

total.add(4);

total.add(2);

total.add(3);

total.add(1);

total.add(5);

total.add(6);

double a = median(total);

System.out.println(a);

}

private static double median(List total) {

double j = 0;

//集合排序

Collections.sort(total);

int size = total.size();

if(size % 2 == 1){

j = total.get((size-1)/2);

}else {

//加0.0是為了把int轉成double類型,否則除以2會算錯

j = (total.get(size/2-1) + total.get(size/2) + 0.0)/2;

}

return j;

}

1. 方法内先判斷集合是奇數還是偶數,如果是奇數那麼就是第n+1/2個數 ,也就是下标為n-1/2的值,

如果是偶數 就是第n/2和n/2+1的數的平均值 也就是下标為n/2-1和n/2的平均值

2. 該方法傳入的是list集合  如果為數組  可以先用Arrays.aslist()方法轉換後傳入