请教个python修改符合条件的元素值的问题

Storm

                            A         B
0 2022-10-01 13:44:23 000
1 2022-10-01 13:44:23 001
2 2022-10-01 13:44:24 550
3 2022-10-01 13:44:24 010
要求: B列>500,则A列的时间加1秒,不符合条件则不修改 请教这个该怎么写呀?我用map,结果把不符合条件的改成nan


def f(x):
    res=str(datetime.datetime.strptime(x,'%Y-%m-%d %H:%M:%S')+datetime.timedelta(seconds=1))
    return res
df1['A']=df1['A'][df1['B']>'500'].apply(f )

Storm

解决了
df2=df1['A'][df1['B']>'500'].map(f)
df1['A']=df1['A'].mask(df1['B']>'500',df2)   

np.where也行:
df1['A']=np.where(df1['B']>'500', df2, df1['A'])