Классификатор: неверный вывод - PullRequest
0 голосов
/ 11 апреля 2019

У меня датафрейм выглядит так:

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"
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...