import?MySQLdb

def?generate_dicts（shop_list）:
????customerid_set=set（）
????for?row?in?shop_list:
????????customerid_set.add（row[0]）
????tempdata_list=[]
????data_set=[]
????for?index?in?customerid_set:
????????????for?row?in?shop_list:
????????????????if（row[0]==index）:
????????????????????tempdata_list.append（row[1]）
????????????data_set.append（tempdata_list）
????????????tempdata_list=[]
????return?data_set
def?load_data_set（）:


????user?=?‘root‘
????pwd?=?‘123456‘
????host?=?‘127.0.0.1‘
????db?=?‘mysql‘
????port?=3306
????try:
????????cnx?=?MySQLdb.connect（host=host?port=port?user=user?passwd=pwd?db=db?）
????????cur?=?cnx.cursor（）
????????cur.execute（“select?customer_idproduct_id?from?shop??“）
????????results?=?cur.fetchall（）
????????shop_list?=?list（results）
????????cur.close（）
????????cnx.close（）
????except?MySQLdb.Errore:
????????print?“MySQL?ERROR?%d:%s“?%（e.args[0]e.args[1]）


????data_set=generate_dicts（shop_list）
????return?data_set


def?create_C1（data_set）:

????C1?=?set（）
????for?t?in?data_set:
????????for?item?in?t:
????????????item_set?=?frozenset（[item]）
????????????C1.add（item_set）
????return?C1


def?is_apriori（Ck_item?Lksub1）:

????for?item?in?Ck_item:
????????sub_Ck?=?Ck_item?-?frozenset（[item]）
????????if?sub_Ck?not?in?Lksub1:
????????????return?False
????return?True


def?create_Ck（Lksub1?k）:

????Ck?=?set（）
????len_Lksub1?=?len（Lksub1）
????list_Lksub1?=?list（Lksub1）
????for?i?in?range（len_Lksub1）:
????????for?j?in?range（1?len_Lksub1）:
????????????l1?=?list（list_Lksub1[i]）
????????????l2?=?list（list_Lksub1[j]）
????????????l1.sort（）
????????????l2.sort（）
????????????if?l1[0:k-2]?==?l2[0:k-2]:
????????????????Ck_item?=?list_Lksub1[i]?|?list_Lksub1[j]
????????????????#?pruning
????????????????if?is_apriori（Ck_item?Lksub1）:
????????????????????Ck.add（Ck_item）
????return?Ck


def?generate_Lk_by_Ck（data_set?Ck?min_support?support_data）:

????Lk?=?set（）
????item_count?=?{}
????for?t?in?data_set:
????????for?item?in?Ck:
????????????