当前位置：首页 > article >正文

PyCharm 中打印完整的 DataFrame

article 2024/12/23 14:45:31

PyCharm 中打印完整的 DataFrame

- 方法 1：设置 `pandas` 的全局显示选项
- - 代码示例：
  - 解释：
- 方法 2：临时设置显示选项
- - 代码示例：
  - 解释：
- 方法 3：使用 `to_string` 方法
- - 代码示例：
  - 解释：
- 总结

在 PyCharm 中打印 DataFrame 时，默认情况下，如果数据行数或列数较多，pandas 会自动省略部分信息。为了显示完整的 DataFrame，你可以通过以下几种方式来调整设置。

方法 1：设置 `pandas` 的全局显示选项

你可以通过修改 pandas 的全局显示选项，来确保在打印 DataFrame 时显示完整的信息。

代码示例：

import pandas as pd

# 设置显示选项
pd.set_option('display.max_rows', None)  # 显示所有行
pd.set_option('display.max_columns', None)  # 显示所有列
pd.set_option('display.width', None)  # 自动调整显示宽度
pd.set_option('display.max_colwidth', None)  # 显示完整的列内容

# 示例 DataFrame
data = {
    'edge_key': ['A', 'A', 'B', 'B', 'C'],
    'source': ['X', 'X', 'Y', 'Y', 'Z'],
    'target': ['P', 'P', 'Q', 'Q', 'R'],
    'text_unit_ids': [[1, 2], [3], [4], [5, 6], [7]],
    'weight': [10, 20, 30, 40, 50],
    'description': ['Edge A', 'Edge A', 'Edge B', 'Edge B', 'Edge C'],
    'human_readable_id': ['A1', 'A2', 'B1', 'B2', 'C1'],
    'id': [1, 2, 3, 4, 5]
}

df = pd.DataFrame(data)
print(df)

解释：

display.max_rows：设置显示的最大行数，None 表示显示所有行。
display.max_columns：设置显示的最大列数，None 表示显示所有列。
display.width：设置显示的宽度，None 表示自动调整。
display.max_colwidth：设置每列的最大宽度，None 表示显示完整内容。

方法 2：临时设置显示选项

如果你不想修改全局设置，可以在打印时临时调整显示选项。

代码示例：

with pd.option_context('display.max_rows', None, 'display.max_columns', None, 'display.max_colwidth', None, 'display.width', None):
    print(df)

解释：

pd.option_context：临时设置显示选项，退出 with 语句块后，选项会恢复为默认值。

方法 3：使用 `to_string` 方法

如果你只想在代码中打印完整的 DataFrame，可以使用 to_string 方法。

代码示例：

print(df.to_string())

解释：

to_string：将 DataFrame 转换为字符串，并显示所有行和列。

总结

通过以上方法，你可以确保在 PyCharm 中打印 DataFrame 时显示完整的信息。以下是几种方法的对比：

方法	适用场景	优点	缺点
全局显示选项	需要长期显示完整信息	一次设置，永久生效	可能影响其他代码的显示效果
临时显示选项	仅在特定代码块中显示完整信息	不影响全局设置	每次都需要手动设置
`to_string` 方法	仅在代码中打印完整信息	简单直接，不影响其他设置	仅适用于打印，不支持格式化