python - 如何直接在 Lambda 中从 pandas 系列映射函数中删除没有结果的记录

Question

import pandas as pd

ser=pd.Series(['Apple','Mango','Cherry','Orange'])

vowels='AEIOUaeiou'

def vowel_check(ser):
    count=0
    for x in ser:
        if x in vowels:
            count+=1
            if count==2:
                return ser
ser=ser.map(lambda x: vowel_check(x))

ser

输出：

0     Apple
1     Mango
2      None
3    Orange
dtype: object

预期的：

0     Apple
1     Mango
3    Orange

我可以删除 None 记录 ser.dropna() 但我们可以直接在 lambda 函数中过滤它们吗？

谢谢

标签： pythonpandas