我正在尝试添加一个新列,将周列转换为该周所属的适当月份。如果一周分为2个月,则以最长月份作为答案。
import pandas as pd
data = {'Week': [201301, 201301, 201302, 201302, 201303],
'product': ['coke', 'fanta', 'coke' ,'sprite' ,'pepsi'],
'quantity': [1.5, 1.7, 3.6, 2.4, 2.9]}
df = pd.DataFrame(data)
print df
Week product quantity
0 201301 coke 1.5
1 201301 fanta 1.7
2 201302 coke 3.6
3 201306 sprite 2.4
4 201307 pepsi 2.9
所以输出看起来像这样。任何想法?
Week product quantity MonthMax
0 201301 coke 1.5 01
1 201301 fanta 1.7 01
2 201302 coke 3.6 01
3 201305 sprite 2.4 02
4 201306 pepsi 2.9 02
即使201301年是12月12日和1月的一部分,201305周也是1月13日和1月13日的一部分。我顺便使用的周格式是yyyyww。我想如果我能算出那个特定周的最后一天,那么月份是多少。但我怎么能以最简单的方式做到这一点。 @alko
答案 0 :(得分:3)
一种可能的方法是将一周的最后一天转换为日期时间,然后访问该月份:
df['LastDayWeek'] = pd.to_datetime((df['Week']-1).astype(str) + "6", format="%Y%U%w")
减号因为python从零开始计数(http://docs.python.org/2/library/datetime.html#strftime-and-strptime-behavior),+ "6"
将是每周的最后一天。
然后你会得到这些日子的月份:
df['MonthMax'] = pd.DatetimeIndex(df['LastDayWeek']).month
给出:
In [6]: df
Out[6]:
Week product quantity LastDayWeek MonthMax
0 201301 coke 1.5 2013-01-05 00:00:00 1
1 201302 fanta 1.7 2013-01-12 00:00:00 1
2 201304 coke 3.6 2013-01-26 00:00:00 1
3 201306 sprite 2.4 2013-02-09 00:00:00 2
4 201308 pepsi 2.9 2013-02-23 00:00:00 2