pandas系列中列表的有效连接

Question

我有以下系列：

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']

谢谢！

Answer 1

嵌套列表理解应该快得多。

>>> [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

Answer 2

我不是在计时或测试这些选项，而是有新的pandas方法explode和numpy.concatenate。