在日-小时映射时旋转时无法排序



我正在尝试像这样标记时间

df['day-hour'] = ('Day' + (df['hour'] // 24).add(1).astype(str) +
' - ' + (df['hour'] % 24).astype(str))

所以,结果将是

customer_id  hour   day-hour
1              10   Day1 - 10
1             123   Day6 - 3
1             489   Day21 - 9
2             230   Day9 - 14

然后我尝试分组df.groupby(['customer_id','day-hour']).size().unstack(fill_value=0)

结果是

day-hour               Day1 - 10   Day6 - 3   Day21 - 9   Day9 - 14
customer_id
1                              1          1           1           0
2                              0          0           0           1

我期望的输出是按实际天数排序的,如下所示

day-hour               Day1 - 10   Day6 - 3   Day9 - 14   Day21 - 9
customer_id
1                              1          1           0           1
2                              0          0           1           0

我应该更改什么代码?

有 2 种可能的解决方案 - 在注释中@Zero添加零:

df['day-hour'] = ('Day' + (df['hour'] // 24).add(1).astype(str).str.zfill(2) +
' - ' + (df['hour'] % 24).astype(str).str.zfill(2) )

或按包含 2 个字段的自定义函数排序:

df = df[sorted(df.columns,key=lambda x: (int(x.split(' - ')[0][3:]), int(x.split(' - ')[1])))]

更好的可读性:

def f(x):
a = x.split(' - ')
return (int(a[0][3:]), int(a[1]))
df = df[sorted(df.columns, key=f)]
print (df)
Day1 - 10  Day6 - 3  Day9 - 14  Day21 - 9
1          1         1          0          1
2          0         0          1          0

最新更新