python - 如何从 pandas 的列表中选择元素？

Question

因此，我创建了一个列来存储包含多个字符串的列表，

import pandas as pd

lists = [['a1','b2'],['a1','b2','c3'],['b2','c3']]
series = pd.Series(lists, index =['XXX', 'YYY', 'ZZZ'])

series

# XXX        [a1, b2]
# YYY    [a1, b2, c3]
# ZZZ        [b2, c3]

我想检查字母是否是字符串的一部分，并且它们根据某些标准进行选择，例如:

if 'b' in series['XXX']:
    series['XXX'] = element containing the selected letter

期望的结果:

XXX| b2         
YYY| [a1, b2, c3]         
ZZZ| [b2, c3]        

有什么方法可以迭代所有行吗？

Answer 1

如果我理解正确的话，您想要列出系列中包含某些字母或字符串的所有元素，例如'b'。不过，这与您的“期望结果”不一致。

但是，这将找到包含“b”的所有元素(并返回一个系列):

string_to_contain = 'b'
series.apply(lambda x: [_ for _ in x if string_to_contain in _])