python: Получить сопоставленные антецеденты по сравнению со списком - PullRequest
Новая
       41

python: Получить сопоставленные антецеденты по сравнению со списком

0 голосов
/ 14 марта 2020

Я пытаюсь получить совпадающие антецеденты из сравнения: список ниже:

********** Отредактировано *********

 dataset = [['Milk', 'Onion', 'Nutmeg', 'Kidney Beans', 'Eggs', 'Yogurt'],

           ['Dill', 'Onion', 'Nutmeg', 'Kidney Beans', 'Eggs', 'Yogurt'],
           ['Milk', 'Apple', 'Kidney Beans', 'Eggs'],
           ['Milk', 'Unicorn', 'Corn', 'Kidney Beans', 'Yogurt'],
           ['Corn', 'Onion', 'Onion', 'Kidney Beans', 'Ice cream', 'Eggs']]

import pandas as pd
from mlxtend.preprocessing import TransactionEncoder
from mlxtend.frequent_patterns import apriori
te = TransactionEncoder()
te_ary = te.fit(dataset).transform(dataset)
df = pd.DataFrame(te_ary, columns=te.columns_)
frequent_itemsets = apriori(df, min_support=0.6, use_colnames=True)
from mlxtend.frequent_patterns import association_rules

rules = association_rules(frequent_itemsets, metric="lift", min_threshold=1.2)

frozenset(rules["antecedents"]).issubset(frozenset(list(["Eggs","Onion"]))) // it is returning False

но он не возвращает true, даже если он совпадает.

в то время как в R, будет выглядеть так: 

rules[which(is.subset(rules@lhs,new_tran)==T)] //its working fine in R

Это я делаю неправильно? Я ценю, если кто-нибудь поможет мне.

...