У меня датафрейм выглядит так:
df
Annotation_A Annotation_B
first Boarding
first Alighting
last Boarding
last Alighting
first Boarding
first Alighting
last Alighting
Я проанализировал данные с этим кодом:
df = pd.read_csv('data.csv')
def classifier(row):
if row["Annotation_A"] == "first" and row["Annotation_B"] == "Boarding":
return "match-1"
elif row["Annotation_A"] == "last" and row["Annotation_B"] == "Alighting":
return "match-2"
else:
return "Un-match"
df["Matching"] = df.apply(classifier, axis=1)
Я хочу создать новый столбец "match", содержащий "match-1 и match-2". Тем не менее, я получил неправильный вывод выглядит так:
Annotation_A Annotation_B Matching
first Boarding match-1
first Alighting Un-match
last Boarding Un-match
last Alighting Un-match -->must be "match-2"
first Boarding match-1
first Alighting Un-match
last Alighting Un-match -->must be "match-2"