如何根据列值Java将2D数组拆分成多个2D数组？

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ArraySplitter {
    public static void main(String[] args) {
        int[][] originalArray = {
                {1, 2, 3},
                {4, 5, 6},
                {7, 8, 9},
                {10, 11, 12}
        };

        int columnToSplit = 0; // 根据第一列的值进行拆分

        Map<Integer, List<int[]>> splitArrays = new HashMap<>();

        for (int[] row : originalArray) {
            int key = row[columnToSplit];

            if (!splitArrays.containsKey(key)) {
                splitArrays.put(key, new ArrayList<>());
            }

            splitArrays.get(key).add(row);
        }

        List<int[][]> resultArrays = new ArrayList<>();

        for (List<int[]> list : splitArrays.values()) {
            int[][] array = new int[list.size()][originalArray[0].length];
            list.toArray(array);
            resultArrays.add(array);
        }

        // 打印拆分后的2D数组
        for (int[][] array : resultArrays) {
            for (int[] row : array) {
                for (int value : row) {
                    System.out.print(value + " ");
                }
                System.out.println();
            }
            System.out.println("------");
        }
    }
}