本文将介绍如何使用 Pandas 库将 DataFrame 从长格式重塑为宽格式。通过 pivot 和 stack 方法的组合，可以灵活地转换数据结构，使得数据更易于分析和可视化。本文提供详细的代码示例和解释，帮助读者掌握 DataFrame 重塑的技巧。
在数据分析中，经常需要对 DataFrame 的结构进行调整，以便更好地适应分析需求。其中，将长格式的数据转换为宽格式是一种常见的操作。Pandas 提供了多种方法来实现这一目标，本文将重点介绍如何使用 pivot 和 stack 方法来实现 DataFrame 的重塑。
使用 pivot 和 stack 重塑 DataFrame
假设我们有以下 DataFrame：
import pandas as pddata = {'Date': [1, 1, 1], 'Item': [1, 2, 3], 'A': ['a1', 'a2', 'a3'], 'B': ['b1', 'b2', 'b3']}df = pd.DataFrame(data)print(df)
输出：
Date Item A B0 1 1 a1 b11 1 2 a2 b22 1 3 a3 b3
我们的目标是将其转换为以下格式：
Date Letter Item 1 Item 2 Item 30 1 A a1 a2 a31 1 B b1 b2 b3
我们可以使用 pivot 方法将 ‘Item’ 列转换为列名，’Date’ 列作为索引，然后使用 stack 方法将列堆叠起来，最后使用 reset_index 方法重置索引。
out = (df.pivot(columns='Item', index='Date').stack(0) .add_prefix('Item ').reset_index(names=['Date', 'Letter']))print(out)
输出：
Date Letter Item 1 Item 2 Item 30 1 A a1 a2 a31 1 B b1 b2 b3
代码解释：
df.pivot(columns=’Item’, index=’Date’): pivot 函数以 ‘Item’ 列的值作为新的列，’Date’ 列作为索引。原始 DataFrame 中的 ‘A’ 和 ‘B’ 列的值将根据 ‘Item’ 列的值进行填充。.stack(0): stack 函数将列索引堆叠到行索引，参数 0 表示堆叠第一层列索引（在本例中是 ‘A’ 和 ‘B’）。.add_prefix(‘Item ‘): 为列名添加 “Item ” 前缀，使列名更清晰。.reset_index(names=[‘Date’, ‘Letter’]): reset_index 函数将索引重置为默认的整数索引，并将原来的索引 ‘Date’ 和堆叠后的 ‘A’ 和 ‘B’ 列命名为 ‘Date’ 和 ‘Letter’。
注意事项：
pivot 函数要求索引和列的组合必须是唯一的。如果存在重复的组合，则会引发错误。stack 函数会将列索引堆叠到行索引，可以指定堆叠的层级。reset_index 函数可以将索引重置为默认的整数索引，并可以指定新的列名。
总结：
通过 pivot 和 stack 方法的组合，可以灵活地将 DataFrame 从长格式重塑为宽格式。这种方法适用于需要将某些列的值转换为列名，并将其他列的值根据这些列名进行填充的场景。掌握这种技巧可以帮助你更好地处理和分析数据。
以上就是使用 Pandas 进行 DataFrame 重塑：从长格式到宽格式的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1370768.html
微信扫一扫 
支付宝扫一扫 
