解压缩NumPy数组

时间:2018-09-10 11:34:04

标签: python arrays numpy

我正在尝试分别打印两个变量的列,这些变量是通过对两个变量进行压缩后应用函数而获得的。请在下面找到:

x5 = list(zip(x3,x4))
x6 = np.array(x5)
print(x6)

应用功能:

n = 1 
br = myfunction(x3,x4,n)

print('this is the result ... ',  br)

请注意,建议的一种可重复的生成myfunction输出类型的数据的方式是:

   br = numpy.random.rand((5, 2))

输出如下所示:

finally result ... [[[5.37692693e+03 3.78425697e- 
 03]
  [1.36092539e+03 3.35722632e-03]
  [5.43477586e+03 3.90104194e-03]
   ...
  [5.93759455e+03 4.05490440e-03]
  [3.00855802e+03 3.65746871e-03]
  [4.69875286e+03 3.93713784e-03]]]

我尝试了许多方法,例如: 首先:

  x10,x11 = zip(*br)
  print(x10,x11) /print(np.array(x10),np.array(x11)) 

没有给出错误提示(期望太多的值来解压两个)

第二:

def column(matrix,i):
    return [row[i] for row in matrix]

 x10 = column(br,0)
 x11 = column(br,1)

 print(x10,x11)

其输出为:

 [array([5.37692693e+03, 3.78425697e-03])] 
 [array([1.36092539e+03, 3.35722632e-03])]

第三个:

   x11 = [i[0] for i in br]
   x12 = [i[1] for i in br]
   print(x11.split(),x12.split())

我认为这不起作用,因为我的函数的输出为[[[..]]]

类型

非常感谢

2 个答案:

答案 0 :(得分:0)

您只能解压缩数组的第一个维,但是要解压缩的维是第二个维。这意味着您可以:

x10, x11 = br.T

请注意,x10x11将是a view into br!这意味着,如果您更改这些数组中任何一个的值,那么您也将更改其他变量。

如果您的数组具有2D以上的尺寸(额外的[]),请使用

检查数组的形状
br.shape

,如果看到多余的空尺寸(1),请使用

删除它们
br = br.squeeze()

答案 1 :(得分:0)

另一个可能的解决方案是

br = [[[5.37692693e+03, 3.78425697e-03],  [1.36092539e+03, 3.35722632e-03],  [5.43477586e+03, 3.90104194e-03],  [5.93759455e+03, 4.05490440e-03],  [3.00855802e+03, 3.65746871e-03],  [4.69875286e+03, 3.93713784e-03]]]

x10,x11 = zip(*br[0])
print(x10,x11) 

输出

(5376.92693, 1360.92539, 5434.77586, 5937.59455, 3008.55802, 4698.75286) (0.00378425697, 0.00335722632, 0.00390104194, 0.0040549044, 0.00365746871, 0.00393713784)