格式化字符串以迭代数据框[重复]

Question

背景：

我有以下代码可以从列表中创建一个数据框：

l = ['the cat meows',
     'the dog barks',
     'the bird chirps']
df = pd.DataFrame(l, columns=['Text'])

输出：

          Text
0   the cat meows
1   the dog barks
2   the bird chirps

所需的输出：

          Text     Animal   
0   the cat meows   cat
1   the dog barks   dog
2   the bird chirps bird

方法：

我尝试使用以下代码获取 Desired Output：

#create list of animal names
animal_list = ['cat', 'dog', 'bird']

#extract names from 'Text' column using the names in 'animal_list' 
#and create a new column containing extracted 'Text' names
df['Sound'] = df['Animal'].str.extract(r"(%s)"% animal_list)

问题：

但是，当我这样做时，我会得到以下信息：

            Text    Animal
0   the cat meows   t
1   the dog barks   t
2   the bird chirps t

问题

如何实现我想要的输出？

Answer 1

将animal_list 与str.extract 一起使用

我们可以在这里使用Series.str.extract 并将animal_list 传递给它，该| 是正则表达式中的or 运算符：

df['Animal'] = df['Text'].str.extract(f"({'|'.join(animal_list)})")

或者如果你有python f-string

我们可以使用来自 cmets 的@Mike 的回答

df['Animal'] = df['Animal'].str.extract(r"({})".format("|".join(animal_list)))

输出

              Text Animal
0    the cat meows    cat
1    the dog barks    dog
2  the bird chirps   bird

用str.split 获取中间词

df['Animal'] = df['Text'].str.split().str[1]

输出

              Text Animal
0    the cat meows    cat
1    the dog barks    dog
2  the bird chirps   bird