如何用Python生成2D高斯？

pip install numpy

import numpy as np

def generate_2d_gaussian(mean, cov, size):
    """
    生成2D高斯分布的矩阵
    :param mean: 均值，形状为(2,)的numpy数组
    :param cov: 协方差矩阵，形状为(2, 2)的numpy数组
    :param size: 生成的矩阵大小，形状为(m, n)的numpy数组
    :return: 生成的2D高斯分布的矩阵
    """
    x, y = np.random.multivariate_normal(mean, cov, size).T
    return np.column_stack((x, y))

# 示例
mean = np.array([0, 0])
cov = np.array([[1, 0], [0, 1]])
size = (1000, 2)

gaussian_matrix = generate_2d_gaussian(mean, cov, size)
print(gaussian_matrix)