如何在pandas dataframe(.to_csv)上面打印标题

时间:2017-04-15 08:33:42

标签: python pandas dataframe

如何在同一个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)

2 个答案:

答案 0 :(得分:2)

如果需要使用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)

答案 1 :(得分:1)

如果在调用.to_csv()函数时未指定#define AES_128_CONF aes_128_driver 参数 - 它将返回一个字符串。因此,您可以简单地将标题与返回的CSV字符串连接起来:

path_or_buf

PS我建议将@jezrael's solution用于大型数据框架,因为我的解决方案可能会导致大型DF f.write("Matrix 1 \n" + oef_df.to_csv())