【Python小技巧】使用prettytable格式化显示dataframe数据

---

前言

经常我们使用print(df)输出dataframe数据，打印输出的数据没有格式，看起来屏幕一篇乱。有没有一种可以格式化输出的工具？

还真有，那就是prettytable。

---

一、安装prettytable

PrettyTable 是 Python 中的一个库，用于以美观的表格形式显示数据。要使用 PrettyTable ，首先需要安装它，可以使用 pip 命令：

pip install prettytable

二、函数打包

def df_table(df):
    from prettytable import PrettyTable
    # 创建 PrettyTable 对象
    table = PrettyTable()

    # 将索引名添加到 PrettyTable
    table.field_names = ["Index"] + list(df.columns)

    # 逐行添加数据，包括索引
    for index, row in df.iterrows():
        table.add_row([index] + list(row))

    # 打印格式化后的表格
    print(table)

三、应用示例

data = {'Name': ['Alice', 'Bob', 'Charlie'],
        'Age': [25, 30, 35],
        'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)
df_table(df)

打印结果如下：

+-------+---------+-----+----------+
| Index |   Name  | Age |   City   |
+-------+---------+-----+----------+
|   0   |  Alice  |  25 | New York |
|   1   |   Bob   |  30 |  London  |
|   2   | Charlie |  35 |  Paris   |
+-------+---------+-----+----------+

---

总结

虽然只是一个简单的方法，但可以使我们的表格清晰的展现出来。对于需要检查数据非常的方便。