赞
踩
# coding=utf-8 import xlrd import json def get_data(dir_case, sheetnum): #如果路径下xlsx文件很多,可以把文件名做一个拼接传入文件名这个参数 # dir_case = 'F:\\code\\csdn\\cese_excel\\' + filename + '.xlsx' ''' 获取其中一张sheet table = data.sheet_by_name(data.sheet_names()[0]) sheet的行数与列数 table.nrows, table.ncols ''' data = xlrd.open_workbook(dir_case) table = data.sheets()[sheetnum] nor = table.nrows nol = table.ncols dict = {} for i in range(0, nor): value_list = [] for j in range(1,nol): title = table.cell_value(i, 0) value = table.cell_value(i, j) if value == "": break # print value value_list.append(value) dict[title] = value_list yield dict with open('LVISandCOCO.txt', 'w') as file: file.write(json.dumps(dict)) ''' • yield 是一个类似 return 的关键字,只是这个函数返回的是个生成器 • 当你调用这个函数的时候,函数内部的代码并不立马执行 ,这个函数只是返回一个生成器对象 • 当你使用for进行迭代的时候,函数中的代码才会执行 ''' if __name__ == '__main__': for m in get_data(r'xxx.xlsx',0): print (m)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。