首页 > 解决方案 > 使用多索引创建具有列名的新 csv 文件

问题描述

我是蟒蛇的新手。在这里,我有以下看起来像的 csv 文件,

Prediction           per_RFC    per_JAPE

     100             100    97.37
    4600             100    86.15
    2200              80      80
    2900             100     100
    160              100     1.36
    3600             100      0
    2000             100      0 

我有特征列

feat_cols =[100,  156,  160,  162,  200,  256,  262,  2000,  2200,   2600,  2900,3600,4600]

在这里,我试图得到一个看起来像的 csv

100  156  160  162  200  256  262  2000  2200   2600  2900       3600     4600
100       100                      100    80           100       100       100 

所以,我试图在这个文件中添加来自 csv 的值

谁能帮我这个 ?

谢谢。

标签: pythonpandasnumpypython-3.7

解决方案


要生成包含预期内容的 DataFrame,请运行:

df2 = df.set_index('Prediction').per_RFC.reindex(feat_cols, fill_value='')\
    .reset_index().T

然后,要将其保存到 CSV 文件,请运行:

df2.to_csv('output.csv', header=False, index=False)

如果您希望按照您所写的方式保存内容(没有逗号,并且列之间有空格),您可以:

  • 生成一个文本变量:tt = df2.to_string(header=False, index=False),
  • 将其保存到输出文件。

推荐阅读