Питон "если не существует, то ..." логика? - PullRequest
2 голосов
/ 24 апреля 2019

С помощью следующего фрейма данных я пытаюсь создать новый столбец guest_1, который принимает первые два слова в каждом элементе столбца guest.Внизу вы можете увидеть мой желаемый результат.

Есть ли какая-то логика "если не существует, тогда ...", которую я могу применить здесь?

Я пробовалСледующее, но очевидное затруднение - это учет человека с одним словом для имени.

df.guest_1 = data.guest.str.split().str.get(0) + ' ' + data.guest.str.split().str.get(1)

df = pd.DataFrame(
    {'date': ['2018-11-21','2018-02-26'],
     'guest': ['Anthony Scaramucci & Michael Avenatti', 'Robyn'],
    })

df.guest_1 = ['Anthony Scaramucci', 'Robyn']

1 Ответ

3 голосов
/ 24 апреля 2019

Вы можете разделить, нарезать и присоединиться.Это изящно обработает ломтики вне границ:

df.guest.str.split().str[:2].str.join(' ')

df['guest_1'] = df.guest.str.split().str[:2].str.join(' ')
df

         date                                  guest             guest_1
0  2018-11-21  Anthony Scaramucci & Michael Avenatti  Anthony Scaramucci
1  2018-02-26                                  Robyn               Robyn
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...