python-3.x - 有没有一种方法可以将列表匹配到其他列表中？

Question

我有一个像这样的数据框：

  target   ;  terms
  Sport    ;  Football,Handball, Tennis
  Mécanique ;  voiture, moto, tracteur, bus
  Technologies;  ordinateur, téléphone,tablette, radio, écouteur

在程序的输入中，我有：

list_terms=  ['Football','Handball','ordinateur','tablette','Tennis']

预期输出：

targets : Sport,Technologies 

标签： python-3.xpandasnlptext-classification

Answer 1

您可以apply()在行上使用并检查terms列在条目列表中是否有任何值。

def check(row):
    targets = [term for term in row['terms'].split(',') if term in entry_list]

    if len(targets) > 0:
        return row['target'] + ' ({:.2f})'.format(len(targets)/len(entry_list))
    else:
        return np.nan

print(df.apply(check, axis=1).dropna().tolist())

"""
['Sport (0.50)', 'Technologies (0.33)']
"""

如果您不想要目标后的概率，您可以使用

def check(row):
    targets = [term for term in row['terms'].split(',') if term in entry_list]
    return len(targets) > 0

print(df[df.apply(check, axis=1)]['target'].tolist())

"""
['Sport', 'Technologies']
"""