pytorch生成csv文件时生成行标题和列标题

可以使用Python中的csv库来生成CSV文件，并在第一行和第一列添加标题。以下是一个示例代码：

import csv
import torch

# 生成数据
data = torch.randn(3, 4)

# 打开CSV文件并写入数据
with open('data.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # 写入列标题
    writer.writerow(['', 'col1', 'col2', 'col3', 'col4'])
    
    # 写入行标题和数据
    for i in range(3):
        row = [f'row{i+1}']
        row.extend(list(data[i]))
        writer.writerow(row)

在此示例中，我们首先生成一个3x4的张量数据。然后，我们打开一个名为“data.csv”的文件，使用csv.writer对象将标题和数据写入文件。在第一行，我们使用writerow方法写入列标题。在接下来的三行中，我们使用一个循环将每一行的行标题和数据写入文件。在行标题中，我们使用f字符串添加行号。在数据中，我们使用list方法将张量转换为列表，然后使用extend方法将数据添加到行中。最后，我们使用writerow方法将整行写入文件。