根据字符串中匹配的日期将两个列表压缩在一起

Question

我有两个从FTP文件夹中提取文件的列表：

sFiles = ftp.nlst(date+'sales.csv')
oFiles = ftp.nlst(date+'orders.csv')

结果显示两个列表如下所示：

^{pr2}$

用我的真实数据集，像。。。在

for sales, orders in zip(sorted(sFiles),sorted(oFiles)): 
     df = pd.concat(...)

获取所需的结果，但有时会出现问题，两个文件都无法放入正确的FTP文件夹中，因此我需要一些代码来创建一个iterable对象，以便根据日期提取匹配的订单和销售文件名。在

下面的工作。。。我不知道我会给它多少“Python”分数。可读性很差，但这是一种理解能力，所以我认为性能会有所提高？在

[(sales, orders) for sales in sFiles for orders in oFiles if re.search(r'\d+',sales).group(0) == re.search(r'\d+',orders).group(0)]

Answer 1

利用熊猫数据帧的索引：

import pandas as pd
sFiles = ['20170822_sales.csv','20170824_sales.csv','20170825_sales.csv','20170826_sales.csv','20170827_sales.csv','20170828_sales.csv']
oFiles = ['20170822_orders.csv','20170823_orders.csv','20170824_orders.csv','20170825_orders.csv','20170826_orders.csv','20170827_orders.csv']

s_dates = [pd.Timestamp.strptime(file[:8], '%Y%m%d') for file in sFiles]
s_df = pd.DataFrame({'sFiles': sFiles}, index=s_dates)

o_dates = [pd.Timestamp.strptime(file[:8], '%Y%m%d') for file in oFiles]
o_df = pd.DataFrame({'oFiles': oFiles}, index=o_dates)

df = s_df.join(o_df, how='outer')

因此：

^{pr2}$