我需要知道房间类型有多少预订。 我有这个DataFrame,每个预订的开始和结束及其房间类型:
date_from date_to room_type
0 2017-08-07 2017-08-12 SUI
1 2017-08-09 2017-08-11 TWN
2 2017-08-09 2017-08-11 QUA
3 2017-08-07 2017-08-11 QUA
4 2017-08-09 2017-08-11 QUA
5 2017-08-09 2017-08-11 QUA
6 2017-08-09 2017-08-11 DBL
7 2017-08-08 2017-08-11 FAM
8 2017-08-08 2017-08-16 INDP
9 2017-08-09 2017-08-11 QUA
我需要这样的东西:
stay_date room_type
0 2017-08-07 SUI
1 2017-08-08 SUI
2 2017-08-09 SUI
3 2017-08-10 SUI
4 2017-08-11 SUI
5 2017-08-09 TWN
5 2017-08-10 TWN
...
所以,我可以像这样创建一个数据透视表:
df = df.pivot_table(
values=['room_type'],
index='stay_date',
aggfunc=len
)
并返回类似的内容:
room_type__code DBL FAM SUI TRP TWIN
stay_date
2017-07-01 61 20 9 19 39
2017-07-02 49 10 7 11 28
2017-07-03 61 14 4 14 40
2017-07-04 65 13 2 11 37
2017-07-05 66 17 2 11 38
答案 0 :(得分:3)
使用itertuples
和date_range
,频率D
与concat
一起创建新的展开DataFrame
:
#convert to datetime if necessary
df['date_from'] = pd.to_datetime(df['date_from'])
#remove one day from to date
df['date_to'] = pd.to_datetime(df['date_to']) - pd.Timedelta(1, unit='d')
df1 = pd.concat([pd.Series(r.room_type,
pd.date_range(r.date_from, r.date_to, freq='D'))
for r in df.itertuples()]) \
.reset_index()
df1.columns = ['stay_date','room_type']
print (df1)
stay_date room_type
0 2017-08-07 SUI
1 2017-08-08 SUI
2 2017-08-09 SUI
3 2017-08-10 SUI
4 2017-08-11 SUI
5 2017-08-09 TWN
6 2017-08-10 TWN
7 2017-08-09 QUA
8 2017-08-10 QUA
9 2017-08-07 QUA
10 2017-08-08 QUA
11 2017-08-09 QUA
12 2017-08-10 QUA
13 2017-08-09 QUA
...
...