用java语言用sort方法实现groupby分组
用排序的方法实现groupby分组，给定数据如下所示。
老王|A|90
李四|B|87
张三|C|70
老王|A|67
张三|A|98
李四|B|76
张三|C|67
张三|C|78
李四|A|89
源程序
import java.io.*;
import java.util.*;
public class DataProcess {
public static void main(String[] args) throws Exception {
String[][] arr = getFile("C:\\Users\\fj-ding\\Desktop\\hw1\\test.txt");
Sortchart sc = new Sortchart();
// 第1列，表示要选择groupby的类别，第2列表示参见计算的数据列，从第0列开始算
int groubyKey = 0;
int dataLine = 2;
sc.process(arr, groubyKey, dataLine);
sc.outputFunction();
System.out.println("我是原矩阵****************************************");
for (int i = 0; i < arr.length; i++) {
for (int j = 0; j < arr[i].length; j++) {
System.out.print(arr[i][j] + " ");
}
System.out.println();
}
}
private static String[][] getFile(String pathName) throws Exception {
File file = new File(pathName);
if (!file.exists())
throw new RuntimeException("Sorry,Not File!");
BufferedReader br = new BufferedReader(new FileReader(file));
String str;
List<String[]> list = new ArrayList<String[]>();
while ((str = br.readLine()) != null) {
int j = 0;
// 正则表达式。
String arr[] = str.split("\|");
String[] dArr = new String[arr.length];
for (String ss : arr) {
dArr[j++] = ss;
}
list.add(dArr);
}
int max = 0;
for (int i = 0; i < list.size(); i++) {
if (max < list.get(i).length)
max = list.get(i).length;
}
// 这个是动态的了，数组长度。size:泛型的个数。
String[][] sanjiaoxing = new String[list.size()][max];
for (int i = 0; i < sanjiaoxing.length; i++) {
for (int j = 0; j < list.get(i).length; j++) {
// 这是一种写法，有点复杂。
sanjiaoxing[i][j] = list.get(i)[j];
}
}
return sanjiaoxing;
}
}
class Sortchart {
String[][] matrixTable;
int column;
String[][] okMatrix;
int dataColumn;
public void process(String[][] matrixTable, int column, int dataColumn) {
int a = 0;
List list0 = new ArrayList();
this.matrixTable = matrixTable;
this.column = column;
this.dataColumn = dataColumn;
for (int i = 0; i < matrixTable.length; i++) {
list0.add(matrixTable[i][column]);
}
Collections.sort(list0);
// 将行向量多加一行，以查询时防止越界。
okMatrix = new String[matrixTable.length + 1][matrixTable[0].length];
for (int j = 0; j < matrixTable.length; j++) {
a = 1;
for (int i = 0; i < matrixTable.length; i++) {
if (matrixTable[i][column] == list0.get(j) && a == 1) {
for (int k = 0; k < matrixTable[0].length; k++) {
okMatrix[j][k] = matrixTable[i][k];
a = 0;
}
}
}
}
// 排列以后的矩阵
System.out.println("我是新矩阵**************************************");
for (int i = 0; i < matrixTable.length; i++) {
for (int j = 0; j < matrixTable[0].length; j++) {
System.out.print(okMatrix[i][j] + " ");
}
System.out.println();
}
}
public void outputFunction() {
// ArrayList list1=new ArrayList();
double[] darr = new double[matrixTable.length];
int count = 0;
double sum = 0;
double avg = 0;
for (int i = 0; i < matrixTable.length; i++) {
for (int j = 0; j < matrixTable[0].length; j++) {
if (okMatrix[i][j].equals(okMatrix[i][column])) {
darr[i] = Double.parseDouble(okMatrix[i][dataColumn]);
// darr[i+1]=Double.parseDouble(okMatrix[i+1][column+1]);
// 小心越界问题
if (okMatrix[i][j].equals(okMatrix[i + 1][j])) {
// sum=darr[i];
sum = sum + darr[i];
count++;
} else {
sum = sum + darr[i];
avg = sum / (count + 1);
// list1.add(sum);
System.out.println("key：" + okMatrix[i][j] + " 求和："
+ sum + " 次数：" + (count+1) + " 均值：" + avg);
sum = 0;
count = 0;
}
}
}
}
}
}
输出结果
我是新矩阵****************************
张三 C 70
张三 A 98
张三 C 67
张三 C 78
李四 B 87
李四 B 76
李四 A 89
老王 A 90
老王 A 67
key：张三 求和：313.0 次数：4 均值：78.25
key：李四 求和：252.0 次数：3 均值：84.0
key：老王 求和：157.0 次数：2 均值：78.5
我是原矩阵*****************************************
老王 A 90
李四 B 87
张三 C 70
老王 A 67
张三 A 98
李四 B 76
张三 C 67
张三 C 78
李四 A 89