在熊猫数据框python上使用Apply函数时如何中断？

Question

有没有一种方法可以使用apply函数为pandas dataframe计算一个新列，但是会根据某些条件“中断”它。

假设我有一个包含2000行的数据框，并且我的标准（取决于先前计算的应用值）在第1000个处变为false 。 计算下一个1000是浪费的，我希望pandas在第1000个之后停止“应用”，并将目前的计算结果还给我。

那可能吗 ？

Answer 1

有条件使用

df['new']=df['old'].apply(lambda x: <calculation> if <condition>(ex : x == 0) else x )

要么

df.loc[df['col']=='value','col']= df.loc[df['col']=='value','col'].apply(lambda x: "new_value"  )

Answer 2

您可以应用在只想要行。

假设您的索引是自动递增且连续的，如果没有，则可以始终执行reset_index来实现。 或者，您可以选择所需的任何条件，而不是前1000行。

idx = df.index < 1000 # index of 1000 items

df.loc[idx, '<yourcolums>'] = df[idx].apply(...)