30

有没有可以调用的方法在 python 中创建随机正交矩阵?可能使用numpy?或者有没有办法使用多个 numpy 方法创建一个正交矩阵?谢谢。

4

7 回答 7

38

scipy 0.18 版具有scipy.stats.ortho_groupscipy.stats.special_ortho_group. 添加它的拉取请求是https://github.com/scipy/scipy/pull/5622

例如,

In [24]: from scipy.stats import ortho_group  # Requires version 0.18 of scipy

In [25]: m = ortho_group.rvs(dim=3)

In [26]: m
Out[26]: 
array([[-0.23939017,  0.58743526, -0.77305379],
       [ 0.81921268, -0.30515101, -0.48556508],
       [-0.52113619, -0.74953498, -0.40818426]])

In [27]: np.set_printoptions(suppress=True)

In [28]: m.dot(m.T)
Out[28]: 
array([[ 1.,  0., -0.],
       [ 0.,  1.,  0.],
       [-0.,  0.,  1.]])
于 2016-07-17T22:35:39.047 回答
23

您可以通过对具有均值和方差的元素 iid 高斯随机变量的矩阵进行分解来获得随机n x n正交矩阵Q,(均匀分布在正交矩阵的流形上) 。这是一个例子:n x nQRn x n01

import numpy as np
from scipy.linalg import qr

n = 3
H = np.random.randn(n, n)
Q, R = qr(H)

print (Q.dot(Q.T))
[[  1.00000000e+00  -2.77555756e-17   2.49800181e-16]
 [ -2.77555756e-17   1.00000000e+00  -1.38777878e-17]
 [  2.49800181e-16  -1.38777878e-17   1.00000000e+00]]

编辑:(在@gg g 发表评论后重新访问此答案。)定理2.3.18-建议了上述关于提供均匀分布(在所谓的Stiefel流形上)正交矩阵的高斯矩阵的QR分解的声明- 19本参考。请注意,结果的陈述表明“类似 QR”的分解,但是,三角矩阵R具有正元素

显然,qrscipy(numpy) 函数的函数并不能保证正对角元素,R并且对应Q的实际上不是均匀分布的。这已在专着 Sec. 中观察到。4.6(讨论指的是 MATLAB,但我猜 MATLAB 和 scipy 都使用相同的 LAPACK 例程)。那里建议通过将其与随机酉对角矩阵后乘来修改由Q提供的矩阵。qr

下面我重现上述参考文献中的实验,绘制由Q提供的“直接”矩阵qr以及“修改”版本的特征值相位的经验分布(直方图),从中可以看出修改后的版本确实有一个均匀的特征值相位,正如从均匀分布的正交矩阵所预期的那样。

from scipy.linalg import qr, eigvals
from seaborn import distplot

n = 50
repeats = 10000

angles = []
angles_modified = []
for rp in range(repeats):
    H = np.random.randn(n, n)
    Q, R = qr(H)
    angles.append(np.angle(eigvals(Q)))
    Q_modified = Q @ np.diag(np.exp(1j * np.pi * 2 * np.random.rand(n)))
    angles_modified.append(np.angle(eigvals(Q_modified))) 

fig, ax = plt.subplots(1,2, figsize = (10,3))
distplot(np.asarray(angles).flatten(),kde = False, hist_kws=dict(edgecolor="k", linewidth=2), ax= ax[0])
ax[0].set(xlabel='phase', title='direct')
distplot(np.asarray(angles_modified).flatten(),kde = False, hist_kws=dict(edgecolor="k", linewidth=2), ax= ax[1])
ax[1].set(xlabel='phase', title='modified');

在此处输入图像描述

于 2016-07-18T07:21:00.087 回答
13

这是rvshttps://github.com/scipy/scipy/pull/5622/files中提取的方法,改动很小 - 足以作为独立的 numpy 函数运行。

import numpy as np    

def rvs(dim=3):
     random_state = np.random
     H = np.eye(dim)
     D = np.ones((dim,))
     for n in range(1, dim):
         x = random_state.normal(size=(dim-n+1,))
         D[n-1] = np.sign(x[0])
         x[0] -= D[n-1]*np.sqrt((x*x).sum())
         # Householder transformation
         Hx = (np.eye(dim-n+1) - 2.*np.outer(x, x)/(x*x).sum())
         mat = np.eye(dim)
         mat[n-1:, n-1:] = Hx
         H = np.dot(H, mat)
         # Fix the last sign such that the determinant is 1
     D[-1] = (-1)**(1-(dim % 2))*D.prod()
     # Equivalent to np.dot(np.diag(D), H) but faster, apparently
     H = (D*H.T).T
     return H

它符合 Warren 的测试,https://stackoverflow.com/a/38426572/901925

于 2016-07-17T23:10:44.877 回答
5

创建任何形状 ( n x m) 正交矩阵的简单方法:

import numpy as np

n, m = 3, 5

H = np.random.rand(n, m)
u, s, vh = np.linalg.svd(H, full_matrices=False)
mat = u @ vh

print(mat @ mat.T) # -> eye(n)

请注意,如果n > m,它将获得mat.T @ mat = eye(m)

于 2019-01-22T11:30:15.387 回答
4

如果您想要一个具有正交列向量的非方阵,您可以使用上述任何一种方法创建一个方阵并删除一些列。

于 2017-12-21T21:04:25.717 回答
4
from scipy.stats import special_ortho_group
num_dim=3
x = special_ortho_group.rvs(num_dim)

文档

于 2019-03-21T21:52:12.900 回答
0

Numpy 也有 qr 分解。https://numpy.org/doc/stable/reference/generated/numpy.linalg.qr.html

import numpy as np

a = np.random.rand(3, 3)
q, r = np.linalg.qr(a)

q @ q.T
# array([[ 1.00000000e+00,  8.83206468e-17,  2.69154044e-16],
#        [ 8.83206468e-17,  1.00000000e+00, -1.30466244e-16],
#        [ 2.69154044e-16, -1.30466244e-16,  1.00000000e+00]])
于 2021-03-31T20:31:46.240 回答