如何将1×3矩阵转换为一维数组？适配点积函数需求

针对你遇到的嵌套矩阵转一维数组的问题，直接提取嵌套结构的第一个元素就能解决，以下是具体实现方式：

原生列表/数组场景 #

如果你的矩阵是纯嵌套列表（比如[[1,2,3]]），在调用点积函数前或者函数内部直接取索引[0]即可：

• 调用前处理：

  # 假设A和B是1×3矩阵
  A_flat = A[0]
  B_flat = B[0]
  dot = 0
  for i in range(len(A_flat)):
      dot += A_flat[i] * B_flat[i]
  

• 函数内部处理（适配两种输入）：

  def dot_product(A, B):
      # 检查并转换嵌套矩阵为一维数组
      if isinstance(A[0], (list, tuple)):
          A = A[0]
      if isinstance(B[0], (list, tuple)):
          B = B[0]
      
      dot = 0
      for i in range(len(A)):
          dot += A[i] * B[i]
      return dot
  

用numpy矩阵的场景 #

如果使用numpy处理矩阵，可以直接用flatten()或ravel()方法转换为一维数组：

import numpy as np

A = np.array([[1,2,3]])
A_flat = A.flatten()  # 得到array([1,2,3])

# 点积计算
dot = np.sum(A_flat * B_flat)
# 或者直接用numpy的dot函数：np.dot(A_flat, B_flat)

核心逻辑就是：1×N矩阵本质是外层只有一个元素的嵌套结构，提取这个唯一的外层元素就能得到一维数组，满足平台要求的dot+=A[i]*B[i]形式。

内容的提问来源于stack exchange，提问作者Skuzzy