Как выбрать элемент из списка в 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 _])