if df column["split"]=='train': code example Example: dataframe split column df[['First','Last']] = df.Name.str.split(" ", expand=True)