如何在Python中生成n个随机元素介于1和30之间且其余元素等于0的矩阵？

import numpy as np

def generate_random_matrix(n):
    # 生成n个随机元素介于1和30之间的一维数组
    random_array = np.random.randint(1, 31, n)
    
    # 创建一个全零矩阵
    matrix = np.zeros((n, n))
    
    # 将随机数组中的元素填充到矩阵的对角线上
    np.fill_diagonal(matrix, random_array)
    
    return matrix

# 示例：生成一个包含5个随机元素的矩阵
random_matrix = generate_random_matrix(5)
print(random_matrix)

[[ 8.  0.  0.  0.  0.]
 [ 0.  3.  0.  0.  0.]
 [ 0.  0.  6.  0.  0.]
 [ 0.  0.  0.  9.  0.]
 [ 0.  0.  0.  0. 12.]]