import pandas as pd
from openpyxl import load_workbook
def df2sheet(dataframe, writer, sname):
dataframe.to_excel(excel_writer=writer, sheet_name=sname,index=None)
writer.save()
writer.close()
def excelAddSheet(df, writer, sname):
book = load_workbook(writer.path)
writer.book = book
df.to_excel(excel_writer=writer,sheet_name=sname,index=None)
writer.close()
import matplotlib.pyplot as plt
excelPath = r"曲面.xlsx"
writer=pd.ExcelWriter(excelPath,engine='openpyxl')
plt.figure(figsize=(20, 10))
for i in range(2, data.shape[1]):
dffA = data[['A量', 'A度']]
dfA = data.iloc[:, i]
dffA = pd.concat([dffA, dfA], axis=1)
alist = cal_lxqm(dffA)
dffA['量'+str(int(i-1))] = alist
sname = '曲面' + str(i-1)
if i == 2:
df2sheet(dffA,writer, sname)
else:
excelAddSheet(dffA, writer, sname)
x = dffA.iloc[:, 0].values
y = dffA.iloc[:, 1].values
z = dffA.iloc[:, 3].values
ax = plt.subplot(2,5,i-1, projection = '3d')
ax.set_title('面' + str(i-1))
ax.scatter(x, y, z, c = 'g', marker = '*', s = 50, alpha = 1)
ax.set_xlabel('A量')
ax.set_ylabel('A度')
ax.set_zlabel('量')
plt.savefig(r'曲面.png', dpi=200)
plt.show();