Сохраните первое слово строки Python?

Question

Скажем, у меня есть список полных имен, как:

names : ["Katie von Martin", "Bob Austin Marley", "John Travolta", "Josh Hartnett"] 

Что бы функция, чтобы разделить каждую строку и сохранить только первое слово выглядит?

def keepFirstName(): 
    for name in names: 
     ???? 

Answer 1

Самый простой будет:

def first_names(names): 
    for name in names: 
     yield name.split()[0] 

например,

>>> print list(first_names(["Katie von Martin", "Bob Austin Marley", "John Travolta", "Josh Hartnett"])) 
['Katie', 'Bob', 'John', 'Josh'] 

Есть несколько обстоятельств, когда вы можете не захотеть разделить строку, если вы просто хотите получить первое слово ... например. если строки ДЕЙСТВИТЕЛЬНО длинный. В этом случае, вы можете использовать str.find, чтобы получить местоположение первого места в строке, а затем вы можете нарезать до этого момента, чтобы дать вам только первое имя:

>>> def first_names(names): 
...  for name in names: 
...   idx = name.find(' ') 
...   yield name[:idx] if idx > 0 else name 
... 
>>> print list(first_names(["Katie von Martin", "Bob Austin Marley", "John Travolta", "Josh Hartnett"])) 
['Katie', 'Bob', 'John', 'Josh'] 

Однако на практике это почти никогда не нужно.

Answer 2

С другой стороны, это даст вам первые слова:

>>> names = ["Katie von Martin", "Bob Austin Marley", "John Travolta", "Josh Hartnett"] 
>>> first_words = [words.split()[0] for words in names] 
>>> print first_words 

['Katie', 'Bob', 'John', 'Josh'] 

Answer 3

сохранить его и хранить его в списке,

b=[] 
for i in names: 
    b.append(i.split()[0]) 

Список б содержит первые имена