python - 如何将数组列表拆分为单独的数组？

Question

我得到了一个数组列表，其中包含两个不同维度的数组。

 c = [array([  3.00493560e+05,   3.04300000e+01,   3.21649837e-01,
         6.50984546e+05,   3.00493379e+05,   3.03073203e+01]), array([ 14.])]

我想根据维度将它们拆分为两个单独的数组。

   a =  array[([  3.00493560e+05,   3.04300000e+01,   3.21649837e-01,
                 6.50984546e+05,   3.00493379e+05,   3.03073203e+01]]

   b = array[([ 14.])]

我尝试使用 np.split(c, 6) - 但它基于给定长度分割数组并创建一个大数组，所以这不是我所期望的。

我也尝试过使用

a = c[c[:, 0] < 1.5]
b = c[c[:, 1] > 5]

它有效，但有时第二个数组中的值与第一个数组中的值具有相同的值...

Answer 1

根据我的理解，您希望将 numpy 数组列表拆分为单独的 python 列表。您可以执行以下操作:

a,b = [ [individualArray] for individualArray in c]

这将为您提供所需的输出:

a= [array([  3.00493560e+05,   3.04300000e+01,   3.21649837e-01,
             6.50984546e+05,   3.00493379e+05,   3.03073203e+01]
b= [array([ 14.])]

编辑

如果 c 包含超过 2 个数组，您可以通过生成分割数组列表来推广此方法:

splitArraysList = [ [individualArray] for individualArray in c ]

如果数组非常大，您可以使用生成器而不是列表来迭代拆分列表中的各个数组:

splitArraysList = ( [individualArray] for individualArray in c )