Selecting first and last row in python

[sudhirkaukuntla]

My code:

def test(x):
    if((df.groupby(['city']).first())== (df.groupby(['city']).last())):
        df["NEW"]=1
    return x
df