python - 将相同键的行组合成单个数组

Question

我有一个 pandas 数据框，如下所示:

    error
 0: [[0.1,0.4,-0.3]]
 1: [[-0.6,-0.3,0.2]]
 .
 .
 .
 99: [[0.4,-0.7,0.1]]

我想将所有值组合到一个数组中，如下所示:

 [0.1,0.4,-0.3,-0.6,-0.3,0.2,...,0.4,-0.7,0.1]

有没有一种快速的方法可以使用 pandas 来执行此操作，或者我是否需要迭代数据并“手动”构建数组？在这种情况下，数据顺序并不重要。

在更一般的情况下，如何组合大小不同的数组(例如第 0 行包含 3 个元素的数组，第 1 行包含 6 个元素的数组等...)？

Answer 1

使用numpy.ravel :

L = np.array(df['error'].values.tolist()).ravel().tolist()
print (L)
[0.1, 0.4, -0.3, -0.6, -0.3, 0.2, 0.4, -0.7, 0.1]

使用str[0]来选择嵌套列表的更通用解决方案:

print (df)
               error
0   [[0.1,0.4,-0.3]]
1      [[-0.6,-0.3]]
99  [[0.4,-0.7,0.1]]

from  itertools import chain

L = list(chain.from_iterable(df['error'].str[0]))
print (L)
[0.1, 0.4, -0.3, -0.6, -0.3, 0.4, -0.7, 0.1]

L =  np.concatenate(df['error'].str[0].values).tolist()
print (L)
[0.1, 0.4, -0.3, -0.6, -0.3, 0.4, -0.7, 0.1]