如何使用 pandas 和 numpy 每天重置一个计数器？

Question

Excel 表中有 2 列（“input.xlsx”）：“时间（分钟）”和“日期”。首先，我想将这些列转换为 numpy 数组，因为我想将这些值用于进一步使用。时间列增加了 15 分钟，我希望每隔 15 分钟计数器列值增加 1。此外，我希望该计数器值在每个新的一天重置。

期望的输出（控制台结果（不是 Xlsx 格式））：

Time(min)        Date       counter  
00:00        27-Aug-18      1
00:15        27-Aug-18      2
00:30        27-Aug-18      3
00:45        27-Aug-18      4
01:00        27-Aug-18      5
01:15        27-Aug-18      6
.
.
.
23:45        27-Aug-18     96
00:00        28-Aug-18     1
00:15        28-Aug-18     2

Answer 1

df['counter'] = 1
df['counter'] = df[['counter','Date']].groupby('Date').transform(lambda x: x.cumsum())

编辑：输出为 numpy 数组

这两个 numpy 数组将是 data 和 counter。

• data - 包含 Time 和 Date
• counter - 包含计数器列

df = pd.read_excel('input.xlsx', header=0)

# Data as it is (DateTime objects and Timestamp objects)
data = df.to_numpy() 
# Or convert all entries to strings
data = df.astype(str).to_numpy()

df['counter'] = 1
counter = df[['counter','Date']].groupby('Date').transform(lambda x: x.cumsum()).to_numpy()