如何基于 string.contains 创建多列

Question

我正在尝试使用以下语句根据单个列中的值创建一个新列/列：

df.loc[df['Figure'].str.contains(r'^(?=.*BLUE)(?=.*HOTEL)'),'Location'] = 'BLUE HOTEL'
df.loc[df['Figure'].str.contains(r'^(?=.*Expensive)'),'Location'] = 'left'
df.loc[df['Figure'].str.contains(r'^(?=.*Cheap)'),'Location'] = 'Right'

如果原始列包含超过 1 个上述字符串，我希望将其插入单独的第二个新列，并可能插入第三个新列。

因此，目标是将确定的关键词分别放入不同的列中。 有人对如何编写此代码有任何建议吗？

提前致谢

Answer 1

通过将所有可能的匹配项放在一个列中，然后在匹配项超过一个时扩展列，容易产生不一致的列，之后难以操作。为 3 个案例中的每一个创建 3 个单独的列将是一个更好的选择。举例说明：

创建示例数据：

data = {'Figure': ['GREAT BLUE HOTEL', 'Expensive choice', 'Cheap and good', 'Nothing interesting', 'Cheap BLUE OCEAN HOTEL']}
df = pd.DataFrame(data)

print(df)

    Figure
0   GREAT BLUE HOTEL
1   Expensive choice
2   Cheap and good
3   Nothing interesting
4   Cheap BLUE OCEAN HOTEL

运行代码：

df['BLUE HOTEL'] = df['Figure'].str.contains(r'^(?=.*BLUE)(?=.*HOTEL)')
df['left'] = df['Figure'].str.contains(r'^(?=.*Expensive)')
df['right'] = df['Figure'].str.contains(r'^(?=.*Cheap)')

结果数据框：

                   Figure  BLUE HOTEL   left  right
0        GREAT BLUE HOTEL        True  False  False
1        Expensive choice       False   True  False
2          Cheap and good       False  False   True
3     Nothing interesting       False  False  False
4  Cheap BLUE OCEAN HOTEL        True  False   True

然后，您可以通过以下方式操作或过滤数据：

df[df['BLUE HOTEL']]        # filter entries with 'BLUE' and 'HOTEL'

Output:
                   Figure  BLUE HOTEL   left  right
0        GREAT BLUE HOTEL        True  False  False
4  Cheap BLUE OCEAN HOTEL        True  False   True


df[df['right']]             # filter entries with 'Cheap'

Output:    
                   Figure  BLUE HOTEL   left  right
2          Cheap and good       False  False   True
4  Cheap BLUE OCEAN HOTEL        True  False   True