首页 > 解决方案 > Python:在 Pandas 的列标题后插入连字符

问题描述

我在 Python Pandas 中创建了一个数据框,如下所示:

import pandas as pd
import os   
cols = ('Name','AGE','SAL')
df = pd.read_csv("C:\\Users\\agupt80\\Desktop\\POC\\Python\\test.csv",names = cols)
print(df)

当我打印数据框时,我得到以下输出:

    Name  AGE  SAL
0   Amit   32  100
1  gupta   33  200
2  hello   34  300
3   Amit   33  100

请帮助让我知道,如何在列标题后插入连字符“-”行,如下所示:

    Name  AGE  SAL
------------------------
0   Amit   32  100
1  gupta   33  200
2  hello   34  300
3   Amit   33  100

标签: pythonpandasdataframe

解决方案


我不知道任何用于打印分隔符的 pandas 自定义选项,但您可以将 df 打印到一个字符串,然后自己插入该行。像这样的东西:

string_repr = df.to_string().splitlines()
string_repr.insert(1, "-" * len(string_repr[0]))
out = '\n'.join(string_repr)


>>> print(out)
    Name  AGE  SAL
------------------
0   Amit   32  100
1  gupta   33  200
2  hello   34  300
3   Amit   33  100

推荐阅读