Python:如何快速堆叠数据框中某一列的所有数组？

Question

如何快速将一列的所有数组快速堆叠成一个大数组？

例如:

Dataframe['Binary_feature'][0] = array([[1,0,0,0,1]]) Dataframe['Binary_feature'][1] = array([[0,1,0,1,0]]) ...... Dataframe['Binary_feature'][i] = array([[0,1,0,1,0]])

如何将“Binary_feature”列中的所有数组堆叠到一个数组中，例如:

array([[1,0,0,0,1], [0,1,0,1,0], ... [0,1,0,1,0]])

Answer 1

您可以squeeze用于删除大小为 1 的维度的数组:

s = pd.Series([np.array([[1,0,0,0,1]]),
               np.array([[0,1,0,1,0]]),
               np.array([[0,1,0,1,0]])])

res = np.array(s.values.tolist()).squeeze()

array([[1, 0, 0, 0, 1],
       [0, 1, 0, 1, 0],
       [0, 1, 0, 1, 0]])

您会发现这比在您的系列上使用 np.vstack 更有效:

s = pd.concat([s]*10000)

assert (np.array(s.values.tolist()).squeeze() == np.vstack(s)).all()

%timeit np.array(s.values.tolist()).squeeze()  # 25.2 ms per loop
%timeit np.vstack(s)                           # 71.9 ms per loop
%timeit np.vstack(s.values)                    # 66.8 ms per loop