使用multiindex用列名创建一个新的csv文件

Question

我是python的新手。在这里，我有以下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中添加值

有人可以帮我吗？

谢谢。

Answer 1

要生成包含所需内容的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)，
• 将其保存到ouptut文件中。