python - 在 Pandas 系列中扩展列表

Question

我有一个看起来像这样的 pandas 系列:

group
A     [1,0,5,4,6,...]
B     [2,2,0,1,9,...]
C     [3,5,2,0,6,...]

我有类似的系列，我想通过扩展每个列表将其添加到现有系列中。我该怎么做？

我试过了

for x in series:
    x.extend(series[series.index[x]])

但这行不通。

Answer 1

考虑系列 s

s = pd.Series([[1, 0], [2, 2], [4, 1]], list('ABC'), name='group')

s

A    [1, 0]
B    [2, 2]
C    [4, 1]
Name: group, dtype: object

您只需添加相似系列即可扩展每个列表。 pandas 将使用底层对象 __add__ 方法来组合成对元素。对于 list，__add__ 方法连接列表。

s + s

A    [1, 0, 1, 0]
B    [2, 2, 2, 2]
C    [4, 1, 4, 1]
Name: group, dtype: object

---

但是，如果元素是 numpy.array

，这将不起作用

s = pd.Series([[1, 0], [2, 2], [4, 1]], list('ABC'), name='group')
s = s.apply(np.array)

在这种情况下，我会确保它们是列表

s.apply(list) + s.apply(list)

A    [1, 0, 1, 0]
B    [2, 2, 2, 2]
C    [4, 1, 4, 1]
Name: group, dtype: object