雪瑞统计

用于使用统计工具和方法的开源 Python 包。

源代码

https://github.com/Xuerui-Yang/xuerui-stat

安装

pip install xuerui-stat

工具和方法

数据管理器

• 描述

它是管理数据文件的工具。导入数据文件后，该工具会自动将其移动到指定的数据目录。并且在导入数据时，该工具可以搜索数据目录中的文件。

• 例子

from xuerui_stat import DataManager
dm = DataManager(enable=True)

该命令定义了一个类来命名工具。首次使用时，用户应使用命令 'set_dir(yourpath)' 和您的首选路径来设置数据目录，例如，

dm.set_dir('/home/xuerui/Documents/Data/')

完成后，目录路径打印如下：

数据目录：'/home/xuerui/Documents/Data/'

类的参数“启用”可以设置为 True 或 False。为 True 时，以下命令导入的数据文件会自动移动到数据目录。

df=dm.import_data('/home/xuerui/Documents/PythonProjects/FactorAnalysis/example.csv')

否则，用户可以使用以下命令手动移动文件：

dm.addto_dir('/home/xuerui/Documents/example.csv')

移动的数据文件通过添加脚本名称来重命名，以便识别它们。例如，使用上述命令将“example.csv”导入“my_example.py”。因此它将被移动到数据目录并重命名为“MyExample_example.csv”。

用户还可以使用检查数据目录下的内容

dm.list_dir()

决策树、绘图树、随机森林

• 描述

这些命令通过使用树相关方法为用户提供数据挖掘工具。

• 例子

如上，以下命令导入要分析的数据。

from xuerui_stat import *
dm = DataManager(enable=False)
data=dm.import_data("/home/xuerui/Documents/PythonProjects/test.csv")

通过指定类别名称，可以将决策树方法应用于数据。

dt=DecisionTree(data,'Cat')
dt.train()
t=dt.tree
print(t)

pt=PlotTree(dt)
pt.tree_structure_plot()

dt.test()
pt.confusion_matrix_plot()

树和混淆矩阵可以通过“PlotTree”模块绘制。

此外，随机森林还可以如下使用：

rf=RandomForest(data,'Cat')
rf.train(num_tree=300,max_depth=0,min_gini=0)
print(rf.oob_error)
dt.test()
print(dt.confusion_matrix)