python - 如何在 pandas 数据框(.to_csv)上方打印标题

Question

如何在同一个 CSV 中打印两个 pandas 数据框上方的不同标题？我尝试了打印命令，但标题是在终端中打印的，而不是 CSV。

if __name__ == "__main__":
 V = result
 W = reference
 H = np.random.random([5,5100])
 basis_mat, coef_mat = nmf_nimfa(V, W, H)

 basis_df = pd.DataFrame(data=basis_mat)
 coef_df = pd.DataFrame(data=coef_mat)

 with open('NMF_nimfa.csv', 'w') as f:
    print "Matrix 1"
    coef_df.to_csv(f)
 with open('NMF_nimfa.csv', 'a') as f:
    print "Matrix 1"
    basis_df.to_csv(f)

Answer 1

如果需要写入带有文本的新行，请使用write:

with open('NMF_nimfa.csv', 'a') as f:
    f.write("Matrix 1 \n")
    coef_df.to_csv(f)
    f.write("Matrix 1 \n")
    basis_df.to_csv(f)