Python/Numpy - Lưu mảng với tiêu đề cột và hàng

Question

Tôi muốn lưu một mảng 2D vào tệp CSV với thông tin "tiêu đề" hàng và cột (như bảng). Tôi biết rằng tôi có thể sử dụng đối số tiêu đề để numpy.savetxt để lưu tên cột, nhưng có cách nào dễ dàng để bao gồm một số mảng (hoặc danh sách) khác làm cột dữ liệu đầu tiên (như tiêu đề hàng) không?

Dưới đây là ví dụ về cách tôi hiện đang thực hiện. Có cách nào tốt hơn để bao gồm những tiêu đề hàng, có lẽ một số trick với savetxt tôi không biết?

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()) 

Answer 1

Có lẽ bạn muốn làm một cái gì đó như thế này:

# Column of row titles 
rows = np.array(['row1', 'row2', 'row3'], dtype='|S20')[:, np.newaxis] 
with open('test.csv', 'w') as f: 
    np.savetxt(f, np.hstack((rows, data)), delimiter=', ', fmt='%s') 

này được ngầm chuyển đổi data tới một mảng của chuỗi, và mất khoảng 200 ms cho mỗi triệu mục trong máy tính của tôi.

Dtype '|S20' có nghĩa là chuỗi gồm hai mươi ký tự. Nếu đó là quá thấp, con số của bạn sẽ được băm nhỏ:

>>> np.asarray([123], dtype='|S2') 
array(['12'], 
    dtype='|S2') 

Một lựa chọn khác, mà từ thử nghiệm hạn chế của tôi là chậm hơn, nhưng cung cấp cho bạn kiểm soát nhiều hơn và không có vấn đề chặt sẽ được sử dụng np.char.mod, như

# Column of row titles 
rows = np.array(['row1', 'row2', 'row3'])[:, np.newaxis] 
str_data = np.char.mod("%10.6f", data) 
with open('test.csv', 'w') as f: 
    np.savetxt(f, np.hstack((rows, str_data)), delimiter=', ', fmt='%s')