Python/Numpy — сохранение массива с заголовками столбцов и строк

Я хочу сохранить двумерный массив в файл CSV с информацией о "заголовке" строки и столбца (например, Таблица). Я знаю, что могу использовать аргумент заголовка для numpy.savetxt, чтобы сохранить имена столбцов, но есть ли простой способ также включить какой-либо другой массив (или список) в качестве первого столбца данных (например, заголовки строк)?

Ниже приведен пример того, как я это делаю в настоящее время. Есть ли лучший способ включить эти заголовки строк, возможно, какой-то трюк с savetxt, о котором я не знаю?

import csv
import numpy as np

data = np.arange(12).reshape(3,4)
# Add a '' for the first column because the row titles go there...
cols = ['', 'col1', 'col2', 'col3', 'col4']
rows = ['row1', 'row2', 'row3']

with open('test.csv', 'wb') as f:
   writer = csv.writer(f)
   writer.writerow(cols)
   for row_title, data_row in zip(rows, data):
      writer.writerow([row_title] + data_row.tolist())