将矩阵数据转储到python中的文本文件时插入行和列标签

Question

我想在 col 和 row 中插入标签。如下所示。

    x   y   z
a   1   3   4
b   4   3   2

其实我可以像下面这样添加标题。

import numpy as np
row_label = ["a", "b"]
col_label = ["x", "y", "z"]
data = np.array([[1,3,4], [4,3,2]])
np.savetxt("out.csv", data, header=",".join(["x", "y", "z"]), delimiter=",", fmt="%.0f", comments='')

out.csv

x,y,z
1,3,4
4,3,2

但是如何也添加列标签？

Answer 1

import numpy as np
import pandas as pd
row_label = ["a", "b"]
col_label = ["x", "y", "z"]
data = np.array([[1,3,4], [4,3,2]])
df = pd.DataFrame(data, index=row_label, columns=col_label)
df.to_csv(r'temp.csv', sep='\t')

temp.csv

    x   y   z
a   1   3   4
b   4   3   2