java 計算中位數方法
阿新 • • 發佈:2019-01-11
最近工作需要 要求把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<Integer> total = new ArrayList<Integer>(); 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<Integer> 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()方法轉換後傳入