Python NumPy中一维数组执行奇异值分解(SVD)的正确姿势

在python中使用numpy进行奇异值分解（svd）时，直接对一维数组（例如形状为(n,)的向量）操作会导致`linalgerror`，因为`numpy.linalg.svd`要求输入至少是二维矩阵。本文将详细讲解此问题的原因，并提供将一维数组重塑为二维矩阵（如(1,n)或(n,1)）的多种解决方案，确保svd顺利执行。

奇异值分解（SVD）是线性代数中一个强大的工具，广泛应用于数据降维、信号处理和推荐系统等领域。在Python中，我们通常使用numpy.linalg.svd函数来执行SVD。然而，当尝试对一个一维数组（例如，一个时间序列或特征向量）执行SVD时，用户可能会遇到LinAlgError。本教程将深入探讨这一问题，并提供实用的解决方案。

1. 理解问题：NumPy SVD对输入维度的要求

许多用户在从MATLAB等环境迁移到Python NumPy时，可能会遇到维度处理上的差异。在MATLAB中，所有数组默认都是至少二维的（例如，一个1xn的行向量或nx1的列向量）。但在NumPy中，数组可以是一维的，其形状表示为(n,)。

numpy.linalg.svd函数明确要求其输入矩阵a至少是二维的。如果传入一个一维数组，例如np.array([1, 2, 3])（形状为(3,)），它将抛出以下错误：

LinAlgError: 1-dimensional array given. Array must be at least two-dimensional

这表明NumPy的SVD实现需要一个具有行和列概念的矩阵作为输入，而不是一个简单的向量。

立即学习“Python免费学习笔记（深入）”；

错误示例：

import numpy as np

# 尝试对一维数组执行SVD
data_1d = np.array([1, 2, 3])
print(f"一维数组形状: {data_1d.shape}")

try:
    U, s, Vt = np.linalg.svd(data_1d)
except np.linalg.LinAlgError as e:
    print(f"发生错误: {e}")

2. NumPy数组的维度与形状

在解决问题之前，理解NumPy中不同数组形状的含义至关重要：

• 一维数组 (n,)： 表示一个包含n个元素的向量，没有明确的行或列概念。例如 np.array([1, 2, 3]) 的形状是 (3,)。
• 二维行向量 (1,n)： 表示一个1行n列的矩阵。例如 np.array([[1, 2, 3]]) 的形状是 (1,3)。
• 二维列向量 (n,1)： 表示一个n行1列的矩阵。例如 np.array([[1], [2], [3]]) 的形状是 (3,1)。

numpy.linalg.svd函数需要输入是(m, n)形状的二维矩阵。

3. 解决方案：将一维数组重塑为二维矩阵

解决此问题的核心是将一维数组转换为二维矩阵。根据您的具体需求，您可以将其重塑为行向量（1xn）或列向量（nx1）。

3.1 重塑为行向量 (1,n)

当您希望将一维数据视为单个观测值（一行）的特征向量时，可以将其重塑为1行n列的矩阵。

方法一：使用双层方括号创建

最直接的方法是将一维数组包装在额外的方括号中，使其成为一个包含一个列表的列表，NumPy会自动将其解释为二维数组。

import numpy as np

data_1d = np.array([1, 2, 3])

# 方法一：使用双层方括号
data_2d_row = np.array([data_1d.tolist()]) # 或直接 np.array([[1,2,3]])
print(f"重塑为行向量 (1,n) 形状: {data_2d_row.shape}")

U_row, s_row, Vt_row = np.linalg.svd(data_2d_row)
print("\nSVD结果 (行向量输入):")
print(f"U 形状: {U_row.shape}\nU:\n{U_row}")
print(f"s 形状: {s_row.shape}\ns:\n{s_row}")
print(f"Vt 形状: {Vt_row.shape}\nVt:\n{Vt_row}")

方法二：使用 np.reshape 或 np.expand_dims

奇域

奇域是一个专注于中式美学的国风AI绘画创作平台

30

查看详情

这两种方法更具通用性，适用于已存在的NumPy一维数组。

import numpy as np

data_1d = np.array([1, 2, 3])

# 方法二：使用 reshape
data_2d_row_reshape = data_1d.reshape(1, -1) # -1 表示根据其他维度自动推断
print(f"重塑为行向量 (1,n) 形状 (reshape): {data_2d_row_reshape.shape}")

# 方法三：使用 np.expand_dims
data_2d_row_expand = np.expand_dims(data_1d, axis=0) # 在第0轴（行）增加一个维度
print(f"重塑为行向量 (1,n) 形状 (expand_dims): {data_2d_row_expand.shape}")

# 验证SVD
U_row_exp, s_row_exp, Vt_row_exp = np.linalg.svd(data_2d_row_expand)
# 结果与上述方法一相同

3.2 重塑为列向量 (n,1)

当您希望将一维数据视为一个具有n个观测值（行）的单一特征时，可以将其重塑为n行1列的矩阵。

方法一：使用 [:, None] 或 np.newaxis

这是将一维数组转换为列向量的NumPy惯用方法。

import numpy as np

data_1d = np.array([1, 2, 3])

# 方法一：使用切片和None（或np.newaxis）
data_2d_col = data_1d[:, None] # 或 data_1d[:, np.newaxis]
print(f"重塑为列向量 (n,1) 形状: {data_2d_col.shape}")

U_col, s_col, Vt_col = np.linalg.svd(data_2d_col)
print("\nSVD结果 (列向量输入):")
print(f"U 形状: {U_col.shape}\nU:\n{U_col}")
print(f"s 形状: {s_col.shape}\ns:\n{s_col}")
print(f"Vt 形状: {Vt_col.shape}\nVt:\n{Vt_col}")

方法二：使用 np.reshape 或 np.expand_dims

import numpy as np

data_1d = np.array([1, 2, 3])

# 方法二：使用 reshape
data_2d_col_reshape = data_1d.reshape(-1, 1) # -1 表示根据其他维度自动推断
print(f"重塑为列向量 (n,1) 形状 (reshape): {data_2d_col_reshape.shape}")

# 方法三：使用 np.expand_dims
data_2d_col_expand = np.expand_dims(data_1d, axis=1) # 在第1轴（列）增加一个维度
print(f"重塑为列向量 (n,1) 形状 (expand_dims): {data_2d_col_expand.shape}")

# 验证SVD
U_col_exp, s_col_exp, Vt_col_exp = np.linalg.svd(data_2d_col_expand)
# 结果与上述方法一相同

4. 不同重塑方式对SVD结果的影响

虽然两种重塑方式都能成功执行SVD，但它们会影响输出矩阵U、s和Vt的形状和解释：

• 输入为行向量 (1,n)：

  • U (左奇异向量矩阵) 的形状将是 (1,1)。
  • s (奇异值数组) 的形状将是 (1,)。
  • Vt (右奇异向量矩阵的转置) 的形状将是 (n,n)。 这种情况下，U 和 s 只包含一个值，而 Vt 包含了所有重要的方向信息。

• 输入为列向量 (n,1)：

  • U (左奇异向量矩阵) 的形状将是 (n,n)。
  • s (奇异值数组) 的形状将是 (1,)。
  • Vt (右奇异向量矩阵的转置) 的形状将是 (1,1)。 这种情况下，U 包含了所有重要的方向信息，而 s 和 Vt 只包含一个值。

选择哪种重塑方式取决于您如何概念化您的数据以及后续分析中需要哪种形式的奇异向量。例如，如果您的数据是单个时间序列，将其视为一个行向量（1xn）可能更符合直觉，其右奇异向量（Vt）将提供该时间序列在不同维度上的模式。

5. 总结与最佳实践

在Python中使用numpy.linalg.svd时，始终牢记其对输入矩阵维度的要求是至少二维。当处理一维数组时，务必将其显式地重塑为二维矩阵，无论是行向量(1,n)还是列向量(n,1)。

• 明确维度： NumPy对数组维度有严格的定义，这与MATLAB等工具的行为有所不同。
• 选择合适的重塑方法： np.reshape(1, -1) 或 np.expand_dims(..., axis=0) 用于创建行向量；arr[:, None] 或 np.reshape(-1, 1) 或 np.expand_dims(..., axis=1) 用于创建列向量。
• 理解SVD输出： 不同的输入形状会导致U、s、Vt的输出形状不同，这会影响您对SVD结果的解释和后续处理。

通过遵循这些指导原则，您可以避免常见的LinAlgError，并有效地在NumPy中执行奇异值分解。

以上就是Python NumPy中一维数组执行奇异值分解(SVD)的正确姿势的详细内容，更多请关注php中文网其它相关文章！