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

Question

我想在列和行中插入标签。如下所示。

    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='')

输出.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')

临时.csv

    x   y   z
a   1   3   4
b   4   3   2