在pandas中搜索多个字符串而无需预定义要使用的字符串数

Question

我想知道下面是否有更通用的方法？我想知道是否有一种方法可以创建st函数，以便我可以搜索未预定义数量的字符串？

因此，例如，能够创建通用的st函数，然后键入st（’Governor’，’Virginia’，’Google）

这是我当前的功能，但是它预定义了您可以使用的两个词。（df是pandas DataFrame）

def search(word1, word2, word3 df):
    """
    allows you to search an intersection of three terms
    """
    return df[df.Name.str.contains(word1) & df.Name.str.contains(word2) & df.Name.str.contains(word3)]

st('Governor', 'Virginia', newauthdf)

Answer 1

您可以使用np.logical_and.reduce：

import pandas as pd
import numpy as np
def search(df, *words):  #1
    """
    Return a sub-DataFrame of those rows whose Name column match all the words.
    """
    return df[np.logical_and.reduce([df['Name'].str.contains(word) for word in words])]   # 2


df = pd.DataFrame({'Name':['Virginia Google Governor',
                           'Governor Virginia',
                           'Governor Virginia Google']})
print(search(df, 'Governor', 'Virginia', 'Google'))

版画

                       Name
0  Virginia Google Governor
2  Governor Virginia Google

1. 将*在def search(df, *words)允许search接受的位置参数的数量不受限制。它将收集所有参数（在第一个参数之后）并将其放在名为的列表中words。
2. np.logical_and.reduce（[X，Y，Z]）等效于X & Y & Z。但是，它允许您处理任意长的列表。