提取numpy数组I[0]后生成H_k数组异常，如何得到预期二维数组？

你的问题根源在于I是二维数组（形状为(5,1)），I[0]取出的是形状为(1,)的一维子数组，而非标量。以下几种方法都能得到预期的H_k：

• 方法1：使用双重索引直接取标量

  i_k = I[0, 0]  # 得到标量0.2
  H_k = np.array([[i_k, 1]])  # 结果为[[0.2 1.]]
  

• 方法2：用item()提取一维数组中的标量值

  i_k = I[0].item()
  H_k = np.array([[i_k, 1]])
  

• 方法3：直接在创建H_k时完成索引，省略中间变量

  H_k = np.array([[I[0, 0], 1]])
  

内容的提问来源于stack exchange，提问作者Nick X Tsui