【机器学习】使用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()