显示来自 dataframe 的行，其中列值（字符串）与输入值（字符串）相似

Question

我有以下 dataframe：

    column1
0   knight
1   chair
2   chicken
3   mango

现在我想在这个 dataframe 中搜索与我给计算机的输入字母相似的单词。 例如：

input = "a"

output= [chair, mango]

或者

input = "kn"

output = [knight]

到目前为止，我所做的只是检查输入值和行值是否相同。 但我不希望它们完全相同，而是相似。

Answer 1

你可以使用contains()

这会将包含您的字符串的行标记为 true，而将相反的字符串标记为 false：

data = df['column1'].str.contains('kn')

要仅获取包含您的字符串的行，请使用以下命令：

df[df['column1'].str.contains('kn')]