import openpyxl # 创建Excel文件 workbook = openpyxl.Workbook() # 创建sheet1 sheet1 = workbook.create_sheet("inputs", 0) for i in range(len(input_test)): # 写入数据 for x in range(6): for y in range(32): for z in range(32): sheet1.cell(row=x*32+y+1, column=z+1, value=inputs[x, y, z]) # 重命名sheet sheet1.title = "test_input_" + str(i+1) # 创建sheet2 sheet2 = workbook.create_sheet("outputs", 1) for i in range(len(output_test)): # 写入数据 for x in range(6): for y in range(32): for z in range(32): sheet2.cell(row=x*32+y+1, column=z+1, value=outputs[x, y, z]) # 重命名sheet sheet2.title = "test_output_" + str(i+1) # 创建sheet3 sheet3 = workbook.create_sheet("labels", 2) for i in range(len(output_test)): # 写入数据 for x in range(6): for y in range(32): for z in range(32): sheet3.cell(row=x*32+y+1, column=z+1, value=labels[x, y, z]) # 重命名sheet sheet3.title = "test_label_" + str(i+1) # 保存Excel文件 workbook.save("output.xlsx") 这段代码中的inputs是numpy数组吗,如果是pytorch数组可以吗
时间: 2023-06-18 14:03:08 浏览: 123
使用Python工具包Openpyxl进行Excel处理
在这段代码中,inputs是一个numpy数组。如果你想使用PyTorch数组,你需要先将numpy数组转换为PyTorch张量,然后再进行操作。你可以使用torch.from_numpy()函数将numpy数组转换为PyTorch张量,例如:
```
import torch
inputs = torch.from_numpy(inputs)
```
然后你就可以使用PyTorch张量进行数据的写入和操作了。
阅读全文