基于python的贝叶斯分类算法（数据集为Iris_data）

#coding:utf-8
import math

Iris_setosa_data=[]
Iris_versicolor_data=[]
Iris_virginica_data=[]
#读取训练数据集，这里我将每种花取前45条数据，剩下的5条数据另外存入测试数据集
def read_train_data(filename):
    f=open(filename,'r')
    all_lines=f.readlines()
    for line in all_lines[0:45]:
        line=line.strip().split(',')
        Iris_setosa_data.append(line[0:4])
        #Iris_setosa_label+=1
    for line in all_lines[51:95]:
        line=line.strip().split(',')
        Iris_versicolor_data.append(line[0:4])
        #Iris_versicolor_label+=1
    for line in all_lines[101:145]:
        line=line.strip().split(',')
        Iris_virginica_data.append(line[0:4])
        #Iris_virginica_label+=1
    return Iris_setosa_data,Iris_versicolor_data,Iris_virginica_data

test_data=[]
#读取测试数据集
def read_test_data(testname):
    f=open(testname,'r')
    all_lines=f.readlines()
    for line in all_lines[0:]:
        line=line.strip().split(',')   #以逗号为分割符拆分列表
        test_data.append(line)
    return test_data

#计算均值和方差
def calculate_junzhi_and_fangcha(train_data):
    x1_sum=0.0
    x2_sum=0.0
    x3_sum=0.0
    x4_sum=0.0

    for x in train_data: #计算各个特征的和