使用 .apply() 迭代 pandas 行

Question

我想遍历 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：Series 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。

感谢您的帮助

Answer 1

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

使用 .apply() 迭代 pandas 行

iterating pandas rows using .apply()

loops

literate-programming

pandas