我有一个大型数据框df
,其中包含日期,格式为%Y-%m-%d
。
df
val date
0 356 2017-01-03
1 27 2017-03-28
2 33 2017-07-12
3 455 2017-09-14
我想创建一个新列YearMonth
,其中包含日期,格式为%Y%m
df['YearMonth'] = df['date'].dt.to_period('M')
但是要花很长时间
答案 0 :(得分:2)
您的解决方案在较大的strftime
中比DataFrame
更快,但是输出却不同-Period
与strings
:
df['YearMonth'] = df['date'].dt.strftime('%Y-%m')
df['YearMonth1'] = df['date'].dt.to_period('M')
print (type(df.loc[0, 'YearMonth']))
<class 'str'>
print (type(df.loc[0, 'YearMonth1']))
<class 'pandas._libs.tslibs.period.Period'>
#[40000 rows x 2 columns]
df = pd.concat([df] * 10000, ignore_index=True)
In [63]: %timeit df['date'].dt.strftime('%Y-%m')
237 ms ± 1.7 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
In [64]: %timeit df['date'].dt.to_period('M')
57 ms ± 985 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
列表理解也很慢:
In [65]: %timeit df['new'] = [str(x)[:7] for x in df['date']]
209 ms ± 2.6 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
另一个亚历山大的解决方案:
In [66]: %timeit df['date'].astype(str).str[:7]
236 ms ± 1.4 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
答案 1 :(得分:0)
如果date
列尚未转换为字符串,则可以将其截断为年和月(即前七个字符)。
df['YearMonth'] = df['date'].astype(str).str[:7]
val date YearMonth
0 356 2017-01-03 2017-01
1 27 2017-03-28 2017-03
2 33 2017-07-12 2017-07
3 455 2017-09-14 2017-09