Apriori算法学习笔记(三)
2017-03-29 20:55
162 查看
Apriori算法学习笔记(三)
Apriori算法的Python实现
from __future__ import print_function import pandas as pd # 频繁规则的产生 # 用于实现L_{k-1}到C_k的连接 def find_freq_set(x, ms): x = list(map(lambda i: sorted(i.split(ms)), x)) l = len(x[0]) r = [] for i in range(len(x)): for j in range(i, len(x)): if x[i][:l - 1] == x[j][:l - 1] and x[i][l - 1] != x[j][l - 1]: r.append(x[i][:l - 1] + sorted([x[j][l - 1], x[i][l - 1]])) return r # 寻找关联规则的函数 def find_rule(data, support, confidence, ms=u'--'): result = pd.DataFrame(index=['support', 'confidence']) # 定义输出结果 support_series = 1.0 * data.sum() / len(data) # 支持度序列 column = list(support_series[support_series > support].index) # 初步根据支持度筛选 k = 0 while len(column) > 1: k = k + 1 column = find_freq_set(column, ms) sf = lambda i: data[i].prod(axis=1, numeric_only=True) # 新一批支持度的计算函数 # 当数据集较大时,可以批量处理数据防止内存不足。 data_2 = pd.DataFrame(list(map(sf, column)), index=[ms.join(i) for i in column]).T support_series_2 = 1.0 * data_2[[ms.join(i) for i in column]].sum() / len(data) # 计算连接后的支持度 column = list(support_series_2[support_series_2 > support].index) # 新一轮支持度筛选 support_series = support_series.append(support_series_2) column2 = [] for i in column: # 遍历可能的推理 i = i.split(ms) for j in range(len(i)): column2.append(i[:j] + i[j + 1:] + i[j:j + 1]) cofidence_series = pd.Series(index=[ms.join(i) for i in column2]) # 定义置信度序列 for i in column2: # 计算置信度序列 cofidence_series[ms.join(i)] = support_series[ms.join(sorted(i))] / support_series[ms.join(i[:len(i) - 1])] for i in cofidence_series[cofidence_series > confidence].index: # 置信度筛选 result[i] = 0.0 result[i]['confidence'] = cofidence_series[i] result[i]['support'] = support_series[ms.join(sorted(i.split(ms)))] result = result.T.sort(['confidence', 'support'], ascending=False) return result
相关文章推荐
- Shell 助力开发效率提升
- Python动态类型的学习---引用的理解
- Python3写爬虫(四)多线程实现数据爬取
- 垃圾邮件过滤器 python简单实现
- 下载并遍历 names.txt 文件,输出长度最长的回文人名。
- 浅析时钟向量算法
- install and upgrade scrapy
- Scrapy的架构介绍
- Centos6 编译安装Python
- 使用Python生成Excel格式的图片
- 让Python文件也可以当bat文件运行
- [Python]推算数独
- Python中zip()函数用法举例
- Python中map()函数浅析
- Python将excel导入到mysql中
- Python在CAM软件Genesis2000中的应用
- JSP笔记:<%--aaa --%>和<!--aaa-->两种注释的区别
- 使用Shiboken为C++和Qt库创建Python绑定