从另一个数据框创建一个数据框（使用数据透视）

Question

我遇到了熊猫问题。我有一个包含三列的数据框：'id1'、'id2'、'amount'。

由此，我想创建另一个数据框，其索引为“id1”，列为“id2”，并且单元格包含相应的“金额”。

举个例子：

import pandas as pd
df = pd.DataFrame([['first_person','first_item',10],['first_person','second_item',6],['second_person','first_item',18],['second_person','second_item',36]],columns = ['id1','id2','amount'])

产生：

     id1              id2             amount
0    first_person     first_item      10
1    first_person     second_item     6
2    second_person    first_item      18
3    second_person    second_item     36

由此我想创建第二个数据框：

                 first_item    second_item
first_person     10            6
second_person    18            36

当然，在发布之前，我已经研究了一段时间，但我为此所做的只是一个双重“for循环”......对于我的数据框的大小来说，这是无法计算的.你知道如何以更 Pythonic 的方式做到这一点吗？ （这显然比'for'循环更有效！）

Answer 1

我认为您可以将pivot 与rename_axis 一起使用（pandas 0.18.0 中的新功能）：

print df
             id1          id2  amount
0   first_person   first_item      10
1   first_person  second_item       6
2  second_person   first_item      18
3  second_person  second_item      36

print df.pivot(index='id1', columns='id2', values='amount')
        .rename_axis(None)
        .rename_axis(None, axis=1)

               first_item  second_item
first_person           10            6
second_person          18           36