Wie eine Konstruktion Code als Funktion

Question

Ich bin neu in der Programmierung und Python schreiben, und möchte das folgende Stück Code als Funktion der ‚def‘ ‚Return‘ Konstruktion mit schreiben:

df.loc[df['DATE_INT'].shift(-1) - df['DATE_INT'] == 1, 'CONSECUTIVE_DAY'] = True 
df.loc[(df['DATE_INT'].shift(-1) - df['DATE_INT'] == 1) | (df['DATE_INT'].shift(1) - df['DATE_INT'] == -1), 'CONSECUTIVE_DAY'] = True 

Mein Versuch liefert ungültige Syntax:

def ConsecutiveID(df, column ='DATE_INT'): 
    return df.loc[df['DATE_INT'].shift(-1) - df['DATE_INT'] == 1, 'CONSECUTIVE_DAY'] = True 
      df.loc[(df['DATE_INT'].shift(-1) - df['DATE_INT'] == 1) | (df['DATE_INT'].shift(1) - df['DATE_INT'] == -1), 'CONSECUTIVE_DAY'] = True 

Mein Ziel ist es letztlich meine ConsecutiveID Funktion wie folgt zu verwenden:

df.groupby(['COUNTY_GEOID_YEAR','TEMPBIN']).apply(ConsecutiveID) 

Ich verwende die Split-Apply-Mähdrescher-Konstruktion. Wo growby meine Daten aufteilt, verwende ich die Funktion, die ich in apply konstruieren möchte.

Meine Hauptfrage ist, wie man schreibt, was ich die ConsecutiveID als eine Funktion genannt habe. Danke für jede Hilfe.

Answer 1

def ConsecutiveID(df): 
    df = df.copy() 
    cond1 = df['DATE_INT'].shift(-1) - df['DATE_INT'] == 1 
    cond2 = df['DATE_INT'].shift(1) - df['DATE_INT'] == -1 

    df.loc[cond1 | cond2, 'CONSECUTIVE_DAY'] = True 

    return df