графа, если другой столбец содержит значение

Question

У меня есть CSV-файл с двумя колонками:

fruits values 
bananas  2 
apples  5 
oranges  4 
pineapples 2 

Я хотел бы суммировать valuesfruits если таковые есть «яблоки» в нем. Она должна возвращать 5 + 2 = 7.

Я попытался это:

sum= 0 
if folha['fruits'].str.contains("apples"): 
    sum=sum+folha['values'].sum() 

Но он возвращает ошибку:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all() 

Любые мысли?

Answer 1

Это довольно просто:

In [10]: df 
Out[10]: 
     fruits values 
0  bananas  2 
1  apples  5 
2  oranges  4 
3 pineapples  2 

In [11]: df[df.fruits.str.contains('apples')]['values'].sum() 
Out[11]: 7