时间索引

王大可2000

已知1.xlsx中有列名为flood_start_date和drought_end_date的列，内容格式为1961-07-28，在2.xlsx中存在列名为Year,Mon,Day,PRE的列，相在需要根据1.xlsx中flood_start_date和drought_end_date的列中的时间找2.xlsx前后两个月的数据，要考虑不同月份的天数差异，该如何运用Python实现

FishC

首先，我们需要使用pandas库读取和处理Excel文件。然后，我们可以使用datetime库来进行日期和时间的计算。

下面是实现这个功能的代码：

1. import pandas as pd
   
2. from datetime import datetime, timedelta
   
3. 
   
4. # 读取1.xlsx中的数据
   
5. df1 = pd.read_excel('1.xlsx')
   
6. 
   
7. # 读取2.xlsx中的数据
   
8. df2 = pd.read_excel('2.xlsx')
   
9. 
   
10. # 转换日期格式为datetime类型
    
11. df1['flood_start_date'] = pd.to_datetime(df1['flood_start_date'])
    
12. df1['drought_end_date'] = pd.to_datetime(df1['drought_end_date'])
    
13. 
    
14. # 创建一个空的DataFrame来存储结果
    
15. result = pd.DataFrame()
    
16. 
    
17. # 遍历1.xlsx中的每一行
    
18. for index, row in df1.iterrows():
    
19.     # 获取起始日期和结束日期
    
20.     start_date = row['flood_start_date']
    
21.     end_date = row['drought_end_date']
    
22. 
    
23.     # 计算前两个月的起始日期和结束日期
    
24.     start_date_prev = start_date - timedelta(days=2 * 30)
    
25.     end_date_prev = end_date - timedelta(days=2 * 30)
    
26. 
    
27.     # 根据前两个月的起始日期和结束日期筛选2.xlsx中的数据
    
28.     filtered_data = df2[(df2['Year'] >= start_date_prev.year) & (df2['Year'] <= end_date_prev.year) &
    
29.                         (df2['Mon'] >= start_date_prev.month) & (df2['Mon'] <= end_date_prev.month) &
    
30.                         (df2['Day'] >= start_date_prev.day) & (df2['Day'] <= end_date_prev.day)]
    
31. 
    
32.     # 将筛选出的数据添加到结果中
    
33.     result = result.append(filtered_data, ignore_index=True)
    
34. 
    
35. # 打印结果
    
36. print(result)

复制代码

请注意，上述代码假设`1.xlsx`和`2.xlsx`文件与代码文件在同一目录下。如果不在同一目录下，请指定正确的文件路径。

此代码通过遍历`1.xlsx`中的每一行，并根据其起始日期和结束日期计算前两个月的起始日期和结束日期。然后，在`2.xlsx`中筛选出处于该时间范围内的数据，并将其添加到结果DataFrame中。最后，打印结果DataFrame。

希望能帮到你！

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。