повторение строк pandas с использованием .apply()

Я хотел перебрать фрейм данных pandas, но по какой-то причине он не работает с методом .apply().


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)

Это дает valueError: истинное значение серии неоднозначно. Используйте a.empty, a.bool(), a.item(), a.any() или a.all().

спасибо за помощь


person bmaster69    schedule 18.11.2020    source источник
comment
@QuangHoang не уверен, что ты имел в виду?   -  person bmaster69    schedule 18.11.2020
comment
Я перепишу вашу функцию apply проще: ABCDEFG = train['Pclass'] == 1.   -  person Quang Hoang    schedule 18.11.2020
comment
Не передавать ось = 1   -  person Raghav Sharma    schedule 18.11.2020
comment
@RaghavSharma Спасибо   -  person bmaster69    schedule 23.11.2020
comment
@QuangHoang Спасибо!   -  person bmaster69    schedule 23.11.2020


Ответы (1)


person    schedule
comment
Спасибо @Mortz, этого пока хватит. - person bmaster69; 23.11.2020