Pandas 不聚合重采样

Question

我已经尝试了很多方法来找到一种方法，该方法允许我在没有任何聚合的情况下获得重采样或组的所有组，例如从：

import pandas as pd
import datetime
import numpy as np

fir_date=pd.to_datetime('1991/12/25')
days = pd.date_range(fir_date, fir_date + datetime.timedelta(100), freq='30T')
data=[]
for i in range(3):
    np.random.seed(seed=i)
    data.append(np.random.randint(1, high=100, size=len(days)))
    
df = pd.DataFrame({'Datetime': days, 'feature1': data[0],'feature2': data[1],'feature3': data[2]})
df = df.set_index('Datetime')
df.index = df.index.map(lambda x: x.replace(second=0,microsecond=0))

我能够通过以下方式获得每组特定的重采样：

df.resample('D').groups

> {Timestamp('1991-12-25 00:00:00', freq='D'): 48, 
> Timestamp('1991-12-26 00:00:00', freq='D'): 96,
> Timestamp('1991-12-27 00:00:00', freq='D'): 144,  
> Timestamp('1991-12-28 00:00:00', freq='D'): 192,  
> Timestamp('1991-12-29 00:00:00', freq='D'): 240,  ...}

输出是一个字典，所以我可以通过以下方式访问特定元素：

df.resample('D').get_group('1991-12-25 00:00:00')

但这似乎不太聪明。

有更好的方法，例如，为每组重采样获取 1 个 DataFrame？？

我知道可以将重采样循环为：

for i in df.resample('D'):
    print(i)
    break

但这不允许我比较不连续的组，或者至少不容易。

有什么解决这个问题的好方法吗？

Answer 1

结合使用选择和unique

import pandas as pd
import datetime
import numpy as np

fir_date=pd.to_datetime('1991/12/25')
days = pd.date_range(fir_date, fir_date + datetime.timedelta(100), freq='30T')
data=[]
for i in range(3):
    np.random.seed(seed=i)
    data.append(np.random.randint(1, high=100, size=len(days)))
    
df = pd.DataFrame({'Datetime': days, 'feature1': data[0],'feature2': data[1],'feature3': data[2]})
df['Days'] = df['Datetime'].dt.date # put in your resample step length here, e.g. days, month, ...
df = df.set_index('Datetime')
df.index = df.index.map(lambda x: x.replace(second=0,microsecond=0))
for i in df['Days'].unique():
    selection = df[df['Days'] = i]

Answer 2

如果需要 DataFrame 列表，请使用列表理解：

dfs = [x for i, x in df.resample('D')]

如果需要字典，可以将 resample 对象转换为 tuple 和 dict:

d = dict(tuple(df.resample('D')))

Pandas 不聚合重采样

Pandas resample without aggregation

python

dataframe

pandas

pandas-groupby

pandas-resample