python - 使用 pandas 创建一个由 DataFrame 中的列表组成的新列

Question

给定以下 DataFrame:

   t
0  3
1  5

我想创建一个新列，其中 wach 条目是一个列表，它是它所在行的函数。特别是它应该有一个包含所有正整数的列表，该整数不大于 列中的条目吨。所以输出应该是:

   t  newCol
0  3  [1,2,3]
1  5  [1,2,3,4,5]

换句话说，我想将 list(range(1,t+1)) 应用于每一行。我知道如何在循环中执行此操作，但有一个很长的 DataFrame，所以我正在寻找速度。谢谢。

Answer 1

这是使用 NumPy 方法的矢量化方法 -

a = df.t.values
idx = a.cumsum()
id_arr = np.ones(idx[-1],dtype=int)
id_arr[idx[:-1]] = -a[:-1]+1
df['newCol'] = np.split(id_arr.cumsum(),idx[:-1])

sample 运行-

In [76]: df
Out[76]: 
   t                 newCol
0  4           [1, 2, 3, 4]
1  3              [1, 2, 3]
2  7  [1, 2, 3, 4, 5, 6, 7]
3  2                 [1, 2]
4  5        [1, 2, 3, 4, 5]
5  3              [1, 2, 3]