问熊猫str.split()函数

EN

import re
# create dictionary to assign the name of the hero with space to the one without
dict_hero = { hero: hero.replace(' ','')  for hero in HeroList if ' ' in hero}
# create the inverse of the previous dictionary, several ways but I choose this one
dict_hero_rev = { hero.replace(' ',''):hero  for hero in HeroList if ' ' in hero}
# now create the pattern and the replacement function to use in str.replace
pat = re.compile('|'.join(dict_hero.keys())) #look for the hero's name in your dict_heor keys
repl = lambda x: dict_hero[x.group()] # replace by the corresponding name in the dict_hero
# work on the column Team A
(df['Team A'].str.replace(pat, repl) #change the one with space to without
             .str.split(' ', expand=True) # split on whitespace and expand to columns
             .replace(dict_hero_rev) # replace the hero's names missing a space by the name with space
              .rename(columns={nb: 'Team A Hero {}'.format(nb+1) for nb in range(5)}))

df = pd.DataFrame({'Team A':['Zeus Odin John Wick Jason Bourne Loki',
                             'Hulk Thor Green Lantern Batman Captain America']})

                                           Team A
0           Zeus Odin John Wick Jason Bourne Loki
1  Hulk Thor Green Lantern Batman Captain America

HeroList = ['Green Lantern', 'Thor', 'Hulk', 'Odin', 'Batman', 
              'Jason Bourne', 'Loki', 'John Wick', 'Zeus', 'Captain America']

  Team A Hero 1 Team A Hero 2  Team A Hero 3 Team A Hero 4    Team A Hero 5
0          Zeus          Odin      John Wick  Jason Bourne             Loki
1          Hulk          Thor  Green Lantern        Batman  Captain America