【无标题】

读写Excel

// An highlighted block
#读取excel数据
import xlrd
import numpy as np

data = xlrd.open_workbook("房价数据.xlsx")#打开excel文件读取数据
print(data.sheet_names())
table = data.sheet_by_index(0)#读取工作表
 # 获取行数
nrows = table.nrows
# 获取列数
ncols = table.ncols
excel_list = [[] for i in range(nrows)]
for i in range(3, nrows):
    excel_list[i-3] = table.row_values(i) # 去掉序号，取四个数据
print(excel_list)
# # # 定义excel_list
# # excel_list = []
#
# # for row in range(1, nrows):
# #     for col in range(ncols):
# #         # 获取单元格数据
# #         cell_value = table.cell(row, col).value
# #         # 把数据追加到excel_list中
# #         excel_list.append(cell_value)
# # print(excel_list)
# for row in range(1, nrows):
#     rows[row] = table.row_values(2)#获取行内容
#
# cols = table.col_values(3)#获取列内容
# datalist = np.array(excel_list)  # rows是数据类型是‘list',转化为数组类型好处理
x=excel_list[:,0:ncols-1]
y=excel_list[:,-1]
print(x),
print(y)



# #读取csv数据
# import csv
# import numpy as np
# # with open('F:/泰坦尼克/train.csv','r') as csvfile:
# with open('test.csv','r') as csvfile:
#     reader = csv.reader(csvfile)
#     rows= [row for row in reader]
# print (rows)#输出所有数据
# data=np.array(rows)#rows是数据类型是‘list',转化为数组类型好处理
# x=data[:,0:3]
# y=data[:,-1]
# print(x),
# print(y)
# print("out0=",type(data),data.shape)
# print("out1=",data)

写入excel

wb = op.load_workbook("xiangguan.xlsx")
sh=wb["Sheet1"]
for i in range(len(res)):
    sh.cell(3+i, 2, res[i])# 这里的索引从一开始
wb.save("xiangguan.xlsx")