如何返回矩阵？

import numpy as np

def return_matrix():
    matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
    return matrix

result = return_matrix()
print(result)

import java.util.Arrays;

public class MatrixExample {
    public static int[][] returnMatrix() {
        int[][] matrix = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
        return matrix;
    }

    public static void main(String[] args) {
        int[][] result = returnMatrix();
        System.out.println(Arrays.deepToString(result));
    }
}