pandas系列中列表的高效串联



我有以下系列:

s = pd.Series([['a', 'b'], ['c', 'd'], ['f', 'g']])
>>> s
0    [a, b]
1    [c, d]
2    [f, g]
dtype: object

连接系列中所有列表的最简单的方法是什么,最好是矢量化的方法,这样我就可以得到:

l = ['a', 'b', 'c', 'd', 'f', 'g']

谢谢!

嵌套列表理解应该更快。

>>> [element for list_ in s for element in list_]
    ['a', 'b', 'c', 'd', 'f', 'g']
>>> %timeit -n 100000 [element for list_ in s for element in list_]
100000 loops, best of 3: 5.2 µs per loop
>>> %timeit -n 100000 s.sum()
100000 loops, best of 3: 50.7 µs per loop

直接访问列表中的值甚至更快。

>>> %timeit -n 100000 [element for list_ in s.values for element in list_]
100000 loops, best of 3: 2.77 µs per loop

我没有计时或测试这些选项,但有新的panda方法explodenumpy.concatenate

最新更新