如何在同一个CSV中打印两个大熊猫数据框上的不同标题?我尝试了打印命令,但标题打印在终端而不是CSV。如何在大熊猫数据框上打印标题(.to_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)