python - 随机对角矩阵

Question

我想创建一个大小为 n 的随机对角矩阵，使得对角元素中的每个元素有 50% 的概率为 -1 和 50% 的概率为 1。对此有什么建议吗？

import numpy as np
diagonal_entries = np.random.randint(low = -1, high = 1, size = n)
D = np.diag(diagonal_entries)

但是，问题在于 `np.random.randint 也包含 0 作为值。我只想要 -1 和 1，不包括 0。

Answer 1

您可以使用np.random.choice 对向量进行采样

import numpy as np
n=100
vec=np.random.choice([-1,1],n)
mat=np.diag(vec)