使用Python中的标头将现有数组保存到csv文件

我想获得一些帮助（以最有效的方式）写入csv文件，这是一个已生成的带有标头的数据集。

到目前为止，这是我的代码

import csv
import numpy as np

data = str(np.random.rand(5, 3))

with open('names.csv', 'w') as csvfile:
    fieldnames = ['epoch', 'loss', 'accuracy']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames, delimiter = ' ')
    writer.writeheader()

    writer.writerows(data)

    csvfile.close()