python - Pandas Dataframe 到 Numpy Vstack 数组的唯一列值

Question

我有一个具有以下结构的数据框：

import numpy as np
import pandas as pd

data = {'Group':['1', '1', '2', '2', '3', '3'], 'Value':[1, 2, 3, 4, 5, 6]} 
df = pd.DataFrame(data)

我需要将该数据框（每个唯一组大约有 4000 个值，以及 1000 个组）转换为一个 numpy 数组，如下所示（应保留顺序）

array([[1, 2],[3, 4],[5,6])

另外：99% 的组具有相同的值计数，但有些具有不同的计数。如果一些填充可能会增加到最大值。计数，这样可以避免我丢失数据。

目前，我遍历 uniqe 'Group' 值并将它们 numpy.vstack 在一起。这是缓慢的，远非优雅。

score 1 · Accepted Answer

IIUC，这只是pivot：

(df.assign(col=df.groupby('Group').cumcount())
  .pivot(index='Group', columns='col', values='Value')
  .values
)

输出：

array([[1, 2],
       [3, 4],
       [5, 6]], dtype=int64)

1 回答 1