python - 如何在数据框中查找字符串并创建一个新字符串？

Question

在 Python 3 和 pandas 中，我使用下面的命令创建一个新的数据框。

在“empresas_brasil”数据框中的“nome_empresa”列中，我查找以“CONSULTORIA”一词开头的所有行。

consult = empresas_brasil[empresas_brasil['nome_empresa'].str.startswith("CONSULTORIA", na=False)]

数据帧的格式如下所示:

empresas_brasil.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 8952217 entries, 0 to 8952216
Data columns (total 4 columns):
Unnamed: 0      int64
cnpj            int64
nome_empresa    object
estado          object
dtypes: int64(2), object(2)
memory usage: 273.2+ MB

但现在我需要创建另一个 daframe，在“nome_empresa”列中的任何位置查找单词“CONSULTORIA”，而不仅仅是开头

请问有人知道我该怎么做吗？

Answer 1

您需要将 startswith 更改为 str.contains :

mask = empresas_brasil['nome_empresa'].str.contains("CONSULTORIA", na=False)
consult = empresas_brasil[mask]