условная лямбда-функция из списка? pandas

Question

Я хотел бы найти средний возраст людей, у которых есть собака, использующая функцию лямбда,

, и до сих пор я сделал это код

pets['age'].mean(lambda x: x in pets['pets'] == 'dog')

... но он продолжает давать мне ошибку значения ..... помощь будет очень ценной!

Answer 1

При необходимости age с собакой и возможными фильтрами других значений по boolean indexing с DataFrame.loc:

out = df.loc[df['pets'].apply(lambda x: 'dog' in x), 'age'].mean()
print (out)
46.75

При необходимости age только с собакой:

out = df.loc[df['pets'].apply(lambda x: x[0] == 'dog'), 'age'].mean()

out = df.loc[df['pets'].str[0] == 'dog', 'age'].mean()
print (out)
55.333333333333336

Answer 2

Вы также можете выполнять эту работу без лямбда-функции. Одно из возможных решений:

df.explode('pets').query('pets == "dog"').age.mean()