使用的库有:
import xlrd # 读取excel数据
import numpy as np #使用np计算数据
import math #数学方法,开方,乘方等
import xlwt ##向excel写入数据, 只能写入新创建的xls表(xlsx不行)
import xlsxwriter ##可以写入xlsx
from xlutils.copy import copy #可以写入现有的excel表
代码:
import xlrd # 导入库,读取数据
import numpy as np
import math
import xlwt ##写入数据,只能写入新的xls表
import xlsxwriter##可以写入xlsx
from xlutils.copy import copy#可以写入现有的工作表入代码片
# 打开文件
xlsx = xlrd.open_workbook('表.xlsx')##没写路径,就是在当前项目文件夹下
sheet1 = xlsx.sheets()[0] # 获得第1张sheet,索引从0开始
sheet1_name = sheet1.name # 获得名称,字符串
sheet1_cols = sheet1.ncols # 获得列数,整型
sheet1_nrows = sheet1.nrows # 获得行数,整型
##将excel数据写入np数组,便于后续操作数据
#num_list= np.zeros((sheet1_nrows,sheet1_cols))
#生成和excel同样大的列表,将excel数据写入列表
num_list=[[0 for i in range(3)]for i in range(sheet1_nrows)]
for i in range(sheet1_nrows) :
for j in range(3):
num_list[i][j] = sheet1.row(i)[j].value###excel数据转化为可计算的数据 data = table.cell(2,0).value
print(num_list)
num_list=np.array(num_list) ##列表转化为numpy数组
#print(num_list[525,2])
##以上将excel导入numpy
# 创建excel文件,用于后续写入excel
workbook = xlwt.Workbook(encoding='ascii')
worksheet = workbook.add_sheet('My Worksheet')
#以下为对数据进行运算
for i in range(sheet1_nrows-1):
##计算距离
a = np.sqrt(np.square(num_list[i+1,0] - num_list[i,0]) + np.square(num_list[i+1,1] - num_list[i,1])+ np.square(num_list[i+1,2] - num_list[i,2])) ##下一时刻x,y,z坐标-上一时刻xyz坐标
print('距离',a)
worksheet.write(i+1, 3, a)#向创建的excel第i+1行,3列写入数据a
workbook.save('表1.xls') # 保存计算的数据为表1
#以下为向已创建的excel中写入数据
wb = copy(xlsx) ##xlsx为之前打开的那个“表.xlsx”,复制一下,赋给wb
s = wb.get_sheet(0)#获取工作簿第一张表
s.write(5,5,'A1') #向5行5列写入“A1”
wb.save('表2.xlsx') #保存为表2,表2包括了原有数据和新数据