我有一个数据框,连续几天从网站上抓取了数百万行广告ID,格式如下:
ad-id Date scraped
SSE-AD-3469148 15-Apr
OAG-AD-12583686 15-Apr
OAG-AD-10476872 15-Apr
SSE-AD-3037758 15-Apr
OAG-AD-12583686 16-Apr
OAG-AD-10476872 16-Apr
SSE-AD-3037758 16-Apr
OAG-AD-10476872 17-Apr
SSE-AD-3037758 17-Apr
我只想输出唯一的广告ID,但我需要知道每个广告ID的开始和结束抓取日期,如下所示:
ad-id First scrape Last Scrape
SSE-AD-3469148 15-Apr 15-Apr
OAG-AD-12583686 15-Apr 16-Apr
OAG-AD-10476872 15-Apr 17-Apr
SSE-AD-3037758 15-Apr 17-Apr
使用Pandas制作此产品的最有效时间是什么?我只能想到涉及几百万行非常慢的循环的方法。
答案 0 :(得分:1)
我只能希望你有真实日期而不是“17-Apr”作为文本值。
首先按日期排序可能最有效,然后在groupby
上对第一个和最后一个值执行ad-id
。
您可以通过选择不对groupby结果进行排序来略微提高速度,例如: df.groupby('ad-id', sort=False)...
df.sort_values('Date scraped', inplace=True)
>>> df.groupby('ad-id')['Date scraped'].agg({'First Scrape': 'first', 'Last Scrape': 'last'})
Last Scrape First Scrape
ad-id
OAG-AD-10476872 17-Apr 15-Apr
OAG-AD-12583686 16-Apr 15-Apr
SSE-AD-3037758 17-Apr 15-Apr
SSE-AD-3469148 15-Apr 15-Apr
有100万条记录的时间
np.random.seed(0)
ad_id = ['SSE-' + str(i) for i in np.random.random_integers(1, 500, 1000000)]
ts = pd.to_datetime(['{0}-{1}-{2}'.format(year, month, day) for year, month, day in zip(years, months, days)])
df = pd.DataFrame({'ad-id': ad_id, 'Date scraped': ts})
%%timeit -n 10
df.sort_values('Date scraped', inplace=True)
df.groupby('ad-id')['Date scraped'].agg({'First Scrape': 'first', 'Last Scrape': 'last'})
10 loops, best of 3: 277 ms per loop
>>> df.groupby('ad-id')['Date scraped'].agg({'First Scrape': 'first', 'Last Scrape': 'last'}).head()
Last Scrape First Scrape
ad-id
SSE-1 2015-12-28 2000-01-02
SSE-10 2015-12-25 2000-01-01
SSE-100 2015-12-25 2000-01-01
SSE-101 2015-12-26 2000-01-05
SSE-102 2015-12-28 2000-01-01
# Slightly faster if you don't sort the results.
%%timeit -n 10
df.sort_values('Date scraped', inplace=True)
df.groupby('ad-id', sort=False)['Date scraped'].agg({'First Scrape': 'first', 'Last Scrape': 'last'})
10 loops, best of 3: 268 ms per loop