python - 将 pandas 系列转换为列表系列或将 numpy 数组转换为列表数组

Question

我有一个系列 s

s = pd.Series([1, 2])

什么是使 看起来像的有效方法

0    [1]
1    [2]
dtype: object

Answer 1

这是一种提取到数组并通过引入新轴 None/np.newaxis 扩展到 2D 的方法-

pd.Series(s.values[:,None].tolist())

这是一个类似的，但通过 reshape 扩展到 2D -

pd.Series(s.values.reshape(-1,1).tolist())

运行时测试使用@P-robot's setup -

In [43]: s = pd.Series(np.random.randint(1,10,1000))

In [44]: %timeit pd.Series(np.vstack(s.values).tolist()) # @Nickil Maveli's soln
100 loops, best of 3: 5.77 ms per loop

In [45]: %timeit pd.Series([[a] for a in s]) # @P-robot's soln
1000 loops, best of 3: 412 µs per loop

In [46]: %timeit s.apply(lambda x: [x]) # @mgc's soln
1000 loops, best of 3: 551 µs per loop

In [47]: %timeit pd.Series(s.values[:,None].tolist()) # Approach1
1000 loops, best of 3: 307 µs per loop

In [48]: %timeit pd.Series(s.values.reshape(-1,1).tolist()) # Approach2
1000 loops, best of 3: 306 µs per loop