如何从熊猫列表中选择元素?

时间:2020-01-15 00:22:06

标签: python python-3.x pandas list dataframe

因此,我创建了一个列,用于存储包含多个字符串的列表,

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]        

有没有办法遍历所有行?

1 个答案:

答案 0 :(得分:0)

如果我理解正确性,您想列出系列中包含某些字母或字符串(例如, 'b'。虽然,这与您的“期望结果”不符。

但是,这将找到所有包含'b'的元素(并返回一个Series):

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