PyCharm 如何美化打印 Pandas 的 DataFrame 数据及相应的包|极客教程

PyCharm 如何美化打印 Pandas 的 DataFrame 数据及相应的包

在进行数据分析和处理时，经常需要对 Pandas 的 DataFrame 数据进行展示和打印。然而，Pandas 默认的打印效果可能不够美观，难以直观地查看数据。在 PyCharm 中，我们可以通过使用一些第三方包来美化打印 Pandas 的 DataFrame 数据，使数据更易读、更清晰。下面将介绍一些常用的包以及如何在 PyCharm 中使用它们来美化打印 DataFrame 数据。

1. tabulate 包

tabulate 是一个用于美化表格输出的 Python 包，可以让表格数据更加易读。我们可以利用 tabulate 包将 Pandas 的 DataFrame 数据以表格形式展示出来，并自定义展示的样式。

安装 tabulate 包：

pip install tabulate

import pandas as pd
from tabulate import tabulate

# 创建一个包含数据的 DataFrame
data = {'Name': ['Alice', 'Bob', 'Carl'],
        'Age': [25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago']}
df = pd.DataFrame(data)

# 使用 tabulate 打印 DataFrame
print(tabulate(df, headers='keys', tablefmt='fancy_grid'))

运行结果：

╒════╤═════════╤═════════╤════════════╕
│    │ Name    │   Age   │ City       │
╞════╪═════════╪═════════╪════════════╡
│  0 │ Alice   │    25   │ New York   │
├────┼─────────┼─────────┼────────────┤
│  1 │ Bob     │    32   │ Los Angeles│
├────┼─────────┼─────────┼────────────┤
│  2 │ Carl    │    28   │ Chicago    │
╘════╧═════════╧═════════╧════════════╛

2. Styler 类

Pandas 提供了一个 Styler 类，可以用于自定义 DataFrame 的样式。我们可以通过设置背景颜色、字体颜色等来美化打印出来的 DataFrame 数据。

# 创建一个包含数据的 DataFrame
data = {'Name': ['Alice', 'Bob', 'Carl'],
        'Age': [25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago']}
df = pd.DataFrame(data)

# 使用 Styler 类美化打印 DataFrame
styler = df.style.set_properties(**{'background-color': 'lightblue',
                                    'color': 'black'}).set_table_styles([{
    'selector': 'th',
    'props': [('background-color', 'pink'), ('color', 'black')]
}])

styler

运行结果：在 PyCharm 中运行上述代码，会得到一个带有自定义样式的 DataFrame。

3. prettytable 包

prettytable 是另一个用于美化表格输出的 Python 包，它可以让表格数据更加易读。我们可以使用 prettytable 包将 Pandas 的 DataFrame 数据以表格形式展示出来。

安装 prettytable 包：

pip install prettytable

from prettytable import PrettyTable

# 创建一个包含数据的 DataFrame
data = {'Name': ['Alice', 'Bob', 'Carl'],
        'Age': [25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago']}
df = pd.DataFrame(data)

# 使用 prettytable 打印 DataFrame
x = PrettyTable()
x.field_names = df.columns

for row in df.itertuples(index=False):
    x.add_row(list(row))

print(x)

运行结果：

+-------+-----+-------------+
|  Name | Age |    City     |
+-------+-----+-------------+
| Alice |  25 |  New York   |
|  Bob  |  32 | Los Angeles |
| Carl  |  28 |  Chicago    |
+-------+-----+-------------+

4. 使用 HTML 渲染

除了使用第三方包来美化打印 Pandas 的 DataFrame 数据，我们还可以通过将数据转换成 HTML 格式并在浏览器中打开来实现美化效果。

# 创建一个包含数据的 DataFrame
data = {'Name': ['Alice', 'Bob', 'Carl'],
        'Age': [25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago']}
df = pd.DataFrame(data)

# 将 DataFrame 转换成 HTML 格式并在浏览器中打开
df_html = df.to_html()
with open('output.html', 'w') as f:
    f.write(df_html)

import webbrowser
webbrowser.open('output.html')