使用日期时间列对pandas数据框进行子集筛选

3

这个问题中,使用idx.min对一个字符串变量和一个日期时间变量进行子集分割。如果要使用两个日期时间变量进行子集分割,应该如何操作? 对于下面的示例数据框,我们如何根据class == C最小基准日期最大date_2日期来筛选行?[答案是第3行]:

print(example)
   slot_id class        day   base_date      date_2
0        1     A     Monday  2019-01-21  2019-01-24
1        2     B    Tuesday  2019-01-22  2019-01-23
2        3     C  Wednesday  2019-01-22  2019-01-24
3        4     C  Wednesday  2019-01-22  2019-01-26
4        5     C  Wednesday  2019-01-24  2019-01-25
5        6     C   Thursday  2019-01-24  2019-01-22
6        7     D    Tuesday  2019-01-23  2019-01-24
7        8     E   Thursday  2019-01-24  2019-01-30
8        9     F   Saturday  2019-01-26  2019-01-31

对于只有 class == "C" 且具有 最小基准日期 的情况,我们可以使用:

df.iloc[pd.to_datetime(df.loc[df['class'] == 'C', 'base_date']).idxmin()]

然而,如果我们有两个或更多带有max/min等条件的日期变量,那么索引解决方案是否仍然可行?使用两个或更多变量进行索引子集是否意味着嵌套df.iloc?这是使用两个或更多日期时间变量进行子集的唯一方法吗?

数据:

print(example.to_dict())
{'slot_id': {0: 1, 1: 2, 2: 3, 3: 4, 4: 5, 5: 6, 6: 7, 7: 8, 8: 9}, 'class': {0: 'A', 1: 'B', 2: 'C', 3: 'C', 4: 'C', 5: 'C', 6: 'D', 7: 'E', 8: 'F'}, 'day': {0: 'Monday', 1: 'Tuesday', 2: 'Wednesday', 3: 'Wednesday', 4: 'Wednesday', 5: 'Thursday', 6: 'Tuesday', 7: 'Thursday', 8: 'Saturday'}, 'base_date': {0: datetime.date(2019, 1, 21), 1: datetime.date(2019, 1, 22), 2: datetime.date(2019, 1, 22), 3: datetime.date(2019, 1, 22), 4: datetime.date(2019, 1, 24), 5: datetime.date(2019, 1, 24), 6: datetime.date(2019, 1, 23), 7: datetime.date(2019, 1, 24), 8: datetime.date(2019, 1, 26)}, 'date_2': {0: datetime.date(2019, 1, 24), 1: datetime.date(2019, 1, 23), 2: datetime.date(2019, 1, 24), 3: datetime.date(2019, 1, 26), 4: datetime.date(2019, 1, 25), 5: datetime.date(2019, 1, 22), 6: datetime.date(2019, 1, 24), 7: datetime.date(2019, 1, 30), 8: datetime.date(2019, 1, 31)}}

数据预处理:

example = pd.DataFrame(example)
example['base_date'] = pd.to_datetime(example['base_date'].astype(str), format='%d%m%Y')
example['base_date'] = example['base_date'].dt.date
example['date_2'] = pd.to_datetime(example['date_2'].astype(str), format='%d%m%Y')
example['date_2'] = example['date_2'].dt.date
1个回答

1
您可以使用transform
yourdf=example[example['base_date']==example.groupby('class')['base_date'].transform('min')]

仅适用于C列

yourdf.loc[yourdf['class']=='C',:]

此外,idxminidxmax 只会返回满足最小或最大条件的第一个索引,因此当存在多个最大或最小值时,它们仍然只显示一个索引。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接