代码之家 › 专栏 › 技术社区 › sunqiang

数据数组的索引问题

numpy python

sunqiang · 技术社区 · 14 年前

例如,有一个矩阵:

import numpy as np
A = np.array([[ 8. , -6. , 2. ], 
              [-0.5, 8. , -6. ], 
              [ 0.5, -0.5, 2. ]])

这是一个LU分解(doolitt_s分解)结果。
我想从A那里得到L和U。
U应该是:

U = np.array([[ 8., -6., 2.], 
              [ 0., 8., -6.], 
              [ 0., 0.,  2.]])

L应该是:

L = np.array([[ 1. , 0. , 0. ], 
              [-0.5, 1. , 0. ], 
              [ 0.5, -0.5, 1.]])

那么,我想知道的是如何从A中得到L和U?

2 回复 | 直到 14 年前

Dzinx 14 年前

您不需要任何索引操作。只使用 tril , triu 和 identity 功能:

import numpy as np
A = np.array([[ 8. , -6. , 2. ], 
              [-0.5, 8. , -6. ], 
              [ 0.5, -0.5, 2. ]])

U = np.triu(A)

#[[ 8. -6.  2.]
# [-0.  8. -6.]
# [ 0. -0.  2.]]

L = np.tril(A, k=-1) + np.identity(3)

#[[ 1.   0.   0. ]
# [-0.5  1.   0. ]
# [ 0.5 -0.5  1. ]]

Josef 14 年前

你想要的在我看来不像是卢分解, http://en.wikipedia.org/wiki/LU_decomposition

>>> U_ = np.array([[ 8., -6., 2.],
              [ 0., 8., -6.],
              [ 0., 0.,  2.]])
>>> L_ = np.array([[ 1. , 0. , 0. ],
              [-0.5, 1. , 0. ],
              [ 0.5, -0.5, 1.]])
>>> np.dot(L_, U_)
array([[  8.,  -6.,   2.],
       [ -4.,  11.,  -7.],
       [  4.,  -7.,   6.]])

lu分解在scipy.linalg中可用

>>> A = np.array([[ 8. , -6. , 2. ], [-0.5, 8. , -6. ], [ 0.5, -0.5, 2. ]])
>>> import scipy.linalg as spla
>>> P, L, U = spla.lu(A)
>>> L
array([[ 1.        ,  0.        ,  0.        ],
       [-0.0625    ,  1.        ,  0.        ],
       [ 0.0625    , -0.01639344,  1.        ]])
>>> U
array([[ 8.        , -6.        ,  2.        ],
       [ 0.        ,  7.625     , -5.875     ],
       [ 0.        ,  0.        ,  1.77868852]])
>>> np.dot(L, U)
array([[ 8. , -6. ,  2. ],
       [-0.5,  8. , -6. ],
       [ 0.5, -0.5,  2. ]])

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前