【机器学习】使用python脚本将鸢尾花数据写入到excel表格

在机器学习中,遇到了鸢尾花的数据,想通过python脚本将数据保存在表格中.
顺便练习下数据保存,记录一下
代码很简单,都有注释.
有问题之处欢迎指正.

import xlwt
from sklearn.datasets import load_iris


class Iris:
    def __init__(self):
        self.iris_dataset = load_iris()

    def write_xls(self):
        xls = xlwt.Workbook()  # 实例化一个workbook对象
        sheet = xls.add_sheet("iris", cell_overwrite_ok=True)  # 添加一个工作表，名字为iris。
        heads = self.iris_dataset['feature_names']  # 获取鸢尾花数据的标头
        ls = 0
        for head in heads:
            sheet.write(0, ls, head)  # 0表示第一行，循环（ls）列写入工作表中
            ls += 1
        datas = self.iris_dataset['data']  # 获取鸢尾花的数据，类型为numpy.ndarray
        i = 1
        # 将数据分两次循环写入到表中，外围循环行
        for list in datas:
            j = 0
            # 内围循环列
            for data in list:
                sheet.write(i, j, data)
                j += 1
            i += 1
        # 保存文件
        xls.save("C:\\Users\\admin\\Desktop\\鸢尾花数据.xls")
        print('write worked')


def main():
    iris = Iris()
    iris.write_xls()


# 将py文件设置成本地文件，当在本文件启动本项目时，先执行main函数，当被当成包调用时，不执行main函数。
if __name__ == "__main__":
    main()