从作为列的字符串列表中删除某些字符串Pandas.DataFram

Question

我有一个pandas.DataFrame：

    index    question_id    tag
    0        1858           [pset3, game-of-fifteen]
    1        2409           [pset4]
    2        4346           [pset6, cs50submit]
    3        9139           [pset8, pset5, gradebook]
    4        9631           [pset4, recover]

我需要从tag列的字符串列表中删除除pset*字符串之外的所有字符串。在

所以我要用这样的话来结束：

^{pr2}$

请问我该怎么做？在

Answer 1

您可以将函数应用于tag系列，该系列只使用以'pset'开头的元素构造列表

df.tag.apply(lambda x: [xx for xx in x if xx.startswith('pset')])

# returns:
0           [pset3]
1           [pset4]
2           [pset6]
3    [pset8, pset5]
4           [pset4]