Найдите строку в столбце и создайте новый список элементов.

Question

У меня есть поезд с кадрами данных с колонкой «Имя».Моя цель - найти в столбце имени строку «Док».Моя конечная цель - добавить новый столбец в мой фрейм данных под названием «Доктор».Значения в столбце будут 1 или 0. Если имя в столбце содержит строку «Doc», значение в столбце «Доктор» должно быть равно 1. Если оно не содержит «Doc», то значение в столбце «Доктор»должно быть 0.

Это код, который у меня есть.Я пытаюсь создать список #, который затем могу добавить в столбец Доктор.

temp_list = []
for i in train:
    if i in train[train["Name"].str.contains("Doc")]:
        temp_list.append("1")
    else:
        temp_list.append("0")

После запуска этой функции возвращается список 1, но нет 0

['1', '1', '1', '1', '1', '1', '1', '1', '1', '1', '1', '1', '1']

Answer 1

Попробуйте это:

temp_list = []
for word in train["Name"]:
  temp_list.append('1') if 'Doc' in word else temp_list.append('0')