Python 如何在数据框（.to_csv）上方打印标题

如何在同一CSV中打印两个熊猫数据帧上方的不同标题？我尝试了打印命令，但标题是在终端而不是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)

---

如果需要使用文本写入文件新行，请使用

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)

---

如果调用函数时未指定

path\u或_buf

参数，它将返回一个字符串。因此，您可以简单地将标题与返回的CSV字符串连接起来：

f.write("Matrix 1 \n" + oef_df.to_csv())

PS我建议用于大型数据帧，因为我的解决方案可能会潜在地导致大型DFs出现

MemoryError