iterating pandas rows using .apply()

Question

I wanted to iterate through the pandas data frame but for some reason it does not work with .apply() method.


train = pd.read_csv('../kaggletrain')

pclass = train['Pclass']

#  pclass has list of data with either 1, 2 or 3.. 
#  so wanted to return if the cell is 1 then return True or everything False

def abc(pclass):
    if pclass == 1:
        return True 
    else: 
        return False 
    

ABCDEFG = train.apply(abc, axis=1)

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

Thank you for your help

I' rewriting your `apply` function in an easier way: `ABCDEFG = train['Pclass'] == 1`. — Quang Hoang, Nov 18 '20 at 15:50

score 1 · Accepted Answer · answered Nov 18 '20 at 16:08

1

ABCDEFG = train[train['pclass']==1]

answered Nov 18 '20 at 16:08

Mortz

4,654
1
19
35

Thank you @Mortz that will do for now – bmaster69 Nov 23 '20 at 11:12

iterating pandas rows using .apply()

1 Answers1