添加一个新列，其中包含满足条件的所有行的值列表

Question

我有一些数据列表，格式如下：

        date           value
'2017-01-01 01:01:01'   99
'2017-01-02 01:01:01'   98
'2017-01-03 01:01:01'   97
....
'2017-01-30 01:01:01'   95
'2017-01-31 01:01:01'   94

我想添加一列，如下所示：

        date          value     list
 ...
 ...
 ...
'2017-01-30 01:01:01'   95  [99,98,97,95]
'2017-01-31 01:01:01'   94  [99,98,97,95,94]

如果日期在相应行中的日期之前的某个时间范围内（例如 5 天），则第三列是第二列中的值列表。

Answer 1

这样的东西应该可以工作......

df = pd.DataFrame({'date': ['2017-01-01 01:01:01', '2017-01-02 01:01:01', '2017-01-03 01:01:01', '2017-01-30 01:01:01', '2017-01-31 01:01:01'],
                   'value': [99,98,97,95,94]})

df['date'] = pd.to_datetime(df['date'])

def get_list(row):

   subset = df[(row['date'] - df['date'] <= pd.to_timedelta('5 days')) & (row['date'] - df['date'] >= pd.to_timedelta('0 days'))]
   return str(subset['value'].tolist())

df['list'] = df.apply(get_list, axis=1)

输出：

              date  value          list
0 2017-01-01 01:01:01     99          [99]
1 2017-01-02 01:01:01     98      [99, 98]
2 2017-01-03 01:01:01     97  [99, 98, 97]
3 2017-01-30 01:01:01     95          [95]
4 2017-01-31 01:01:01     94      [95, 94]