python - 生成随机转移概率矩阵python

Question

我想知道在给定一些条件的情况下，是否有一种简单的方法可以在 python 中生成一个填充有随机数的方阵:

• 行总和必须为 1。
• 对角线上的值应该明显高于其他值(value)观。

这是一个随机矩阵，生成一个是可能的，但棘手的部分是关于对角线的条件。例如对于 4x4 矩阵，输出应如下所示:

[[0.90, 0.03, 0.03, 0.04],
[0.01, 0.98, 0.005, 0.005],
[0.04, 0.01, 0.92, 0.03],
[0.00, 0.02, 0.03, 0.95]]

有没有好的方法来生成可变大小的矩阵？

Answer 1

这是一个快速而肮脏的解决方案

import random

k = 4

result = [[random.uniform(0, 0.1 / k) for i in range(k)] for j in range(k)]
for j, r in enumerate(result):
    r[j] += 1 - sum(r)

您可能想考虑使用不同的随机分布，也可以看看 numpy。