Pandas DataFrame 透视操作：实现期望的行列转换

本文介绍了如何使用 Pandas 库中的 pivot 方法以及 set_index、T（转置）和 reset_index 等方法组合，将 DataFrame 转换为期望的行列结构。通过实际示例和代码演示，详细讲解了透视操作的步骤和关键参数，帮助读者掌握 DataFrame 数据重塑的技巧。

Pandas DataFrame 透视操作详解

Pandas 提供了强大的数据处理功能，其中透视（pivot）操作是数据重塑的重要手段。当我们需要将 DataFrame 的行转换为列，或者将列转换为行时，透视操作就显得尤为重要。本文将详细介绍如何使用 Pandas 实现 DataFrame 的透视，并解决一些常见的透视问题。

基础透视操作：pivot()

pivot() 方法是 Pandas 中进行透视操作的基础。它的主要参数包括：

• index: 用于作为新 DataFrame 的索引的列名。
• columns: 用于作为新 DataFrame 的列名的列名。
• values: 用于填充新 DataFrame 值的列名。

然而，直接使用 pivot() 方法可能无法得到期望的结果，例如，当 index 参数设置为 None 时，会出现 NaN 值填充的情况。

高级透视操作：set_index().T.reset_index()

为了更灵活地控制透视过程，我们可以结合使用 set_index()、T（转置）和 reset_index() 方法。

1. set_index(column_name): 将指定的列设置为 DataFrame 的索引。
2. .T: 对 DataFrame 进行转置，即行列互换。
3. reset_index(drop=True): 重置索引，drop=True 表示丢弃原索引。

通过这些操作的组合，我们可以实现更精细化的数据重塑。

行者AI

行者AI绘图创作，唤醒新的灵感，创造更多可能

100

查看详情

示例代码

以下代码演示了如何使用 set_index().T.reset_index() 实现期望的透视效果：

import pandas as pd

# 原始 DataFrame
df2 = pd.DataFrame({
    'nombreNumeroUnico': ['UP2_G1_B', 'UP2_G2_B'],
    'pMax': [110.0, 110.0]
})

# 透视操作
result_df = df2.set_index('nombreNumeroUnico').T.reset_index(drop=True)
result_df.columns.name = None  # 移除列名名称
print(result_df)

代码解释:

• df2.set_index('nombreNumeroUnico'): 将 'nombreNumeroUnico' 列设置为索引。
• .T: 对结果进行转置，将索引变为列，列变为索引。
• reset_index(drop=True): 重置索引，丢弃原索引，生成默认的数字索引。
• result_df.columns.name = None: 移除列名的名称，使输出更简洁。

输出结果:

   UP2_G1_B  UP2_G2_B
0     110.0     110.0

注意事项

• 在进行透视操作前，需要仔细分析数据结构，选择合适的列作为索引、列名和值。
• 当数据中存在重复的索引和列名组合时，pivot() 方法会抛出异常。此时，可以考虑使用 pivot_table() 方法，它可以处理重复数据，并支持聚合函数。
• set_index().T.reset_index() 是一种常用的透视技巧，可以灵活地控制透视过程。
• 确保数据类型正确，避免因数据类型不一致导致透视失败。

总结

通过本文的介绍，我们学习了如何使用 Pandas 进行 DataFrame 的透视操作。pivot() 方法是基础，而 set_index().T.reset_index() 则提供了更灵活的控制方式。掌握这些技巧，可以帮助我们更好地进行数据重塑和分析。在实际应用中，需要根据具体的数据结构和需求，选择合适的透视方法，并注意数据类型和重复值等问题。

以上就是Pandas DataFrame 透视操作：实现期望的行列转换的详细内容，更多请关注php中文网其它相关文章！