对 pandas 数据透视表执行分析时，如何创建新数据框？答案

【问题标题】：How do I create a new data frame when performing analysis on a pandas pivot table?对 pandas 数据透视表执行分析时，如何创建新数据框？
【发布时间】：2020-08-03 09:48:36
【问题描述】：

我正在尝试清理我的数据框，以便创建一个显示随时间变化的利润的图表。我将我的数据框按Symbol 分组，认为正确应用分析会更容易，但我不确定这里的逻辑。我想创建一个单独的数据框，如下例所示。

代码：

import pandas as pd

data = {'Action': ['BUY_TO_OPEN', 'SELL_TO_CLOSE', 'BUY_TO_OPEN', 'SELL_TO_CLOSE', 'BUY_TO_OPEN', 'SELL_TO_CLOSE', 'BUY_TO_OPEN', 'SELL_TO_CLOSE', 'BUY_TO_OPEN', 'SELL_TO_CLOSE'],
        'Date': ['2020-07-23', '2020-07-29', '2020-06-04', '2020-06-24', '2020-07-17', '2020-07-21', '2020-05-28', '2020-05-28', '2020-06-29', '2020-07-20'],
        'Quantity': [200.0, 200.0, 130.0, 130.0, 100.0, 100.0, 100.0, 100.0, 120.0, 120.0],
        'Symbol': ['ACHV', 'ACHV', 'ACST', 'ACST', 'AGE', 'AGE', 'AIKI', 'AIKI', 'AIKI', 'AIKI'],
        'tot_value': [-96.16, 163.81, -100.2, 83.07, -100.08, 149.9, -74.08, 71.91, -100.9, 153.48]}

df_trade = pd.DataFrame(data)

df_trade.Date = pd.to_datetime(df_trade.Date)

df_trades = pd.pivot_table(df_trades, index=['Symbol', 'Date', 'Action', 'Quantity'], values=['tot_value'])

print(df_trades)

数据透视表：

                                         tot_value
Symbol Date       Action        Quantity
ACHV   2020-07-23 BUY_TO_OPEN   200.0        -96.16
       2020-07-29 SELL_TO_CLOSE 200.0        163.81
ACST   2020-06-04 BUY_TO_OPEN   130.0       -100.20
       2020-06-24 SELL_TO_CLOSE 130.0         83.07
AGE    2020-07-17 BUY_TO_OPEN   100.0       -100.08
       2020-07-21 SELL_TO_CLOSE 100.0        149.90
AIKI   2020-05-28 BUY_TO_OPEN   100.0        -74.08
                  SELL_TO_CLOSE 100.0         71.91
       2020-06-29 BUY_TO_OPEN   120.0       -100.90
       2020-07-20 SELL_TO_CLOSE 120.0        153.48

我试图实现的数据框示例：

其中日期是 SELL_TO_CLOSE 日期，Profit 列是 tot_value 的总和，使用每个 BUY_TO_OPEN 和 SELL_TO_CLOSE 对进行计算。

Symbol Date          Profit
ACHV   2020-07-29    67.65
ACST   2020-06-24   -17.13
AGE    2020-07-21    49.82
AIKI   2020-05-28    -2.17
AIKI   2020-07-20    52.58

【问题讨论】：

您是否尝试过在 df_trades 数据帧上使用 [groupby] (pandas.pydata.org/pandas-docs/stable/reference/api/…)？

标签： python pandas

【解决方案1】：

从您的数据透视表中，您可以使用您的数据按对排序的事实，通过将 Action 为 BUY_TO_OPEN 和 cumsum 分组来实现这一点

print (df_trades.reset_index()
                .groupby((df_trades.index.get_level_values('Action')
                          =='BUY_TO_OPEN').cumsum())
                .agg({'Symbol':'first', 'Date':'last','tot_value':'sum'})
                .rename(columns={'tot_value':'Profit'})
      )
  Symbol        Date  Profit
1   ACHV  2020-07-29   67.65
2   ACST  2020-06-24  -17.13
3    AGE  2020-07-21   49.82
4   AIKI  2020-05-28   -2.17
5   AIKI  2020-07-20   52.58

【讨论】：