python - 从列中删除空格上的行

Question

我想保留列中的所有行，其中包含单个单词，其余行包含多个要删除的空格。

我的数据框 df 是:

df['drug']
gilenya
fingolimod
ocrevus
dont want in the column
remove this drug row
text mining for drug column

我想创建一个仅包含正确药物词并删除垃圾的新数据框。我已经尝试过以下解决方案，但它给了我一个空白/空的药物列。

df_drug = pd.DataFrame(columns = ['drug'])
df_drug = df_drug[df_drug.drug.str.count(' ')==1]
or, df_drug = df_drug[df_drug.drug.str.contains('')]

您能帮我找到正确的解决方案吗？喜欢:

df_drug.head()
drug
gilenya
fingolimod
ocrevus

Answer 1

您可以使用 lambda 函数返回一个 bool 序列，该序列仅在 df_drug['drug'] 为一个单词时为 True，然后使用该函数从 df_drug 中进行选择系列。

df_drug = df_drug[df_drug['drug'].apply(lambda x: True if len(x.split()) == 1 else False)]

如果您的 drug 列中有 nan，您可能需要向该 lambda 函数添​​加类似 str(x) 的内容。