python - 为每组连续增加的日期添加行号列

Question

我有 DataFrame，其中包含包含日期的列和包含一些值的其他列，比方说，前 100 行根据日期按顺序排列，从 101 到 200 又是相同的日期，只有不同的值，依此类推。我想添加一列，对从 1 到 100 的行进行计数，并在日期重复时从 1 重新开始。示例

Date       | Value | RowNum
2000-01-01 |   2   |    1  
2000-02-01 |   10  |    2
.
.
.
2003-12-01 |  11   |    100
2000-01-01 |  32   |    1  
2000-02-01 |  14   |    2
.
.
.
2003-12-01 |  4    |    100

我需要这个来透视这个表，其中列是日期，值是值，RowNum 将是索引。谢谢您的帮助。

Answer 1

如果完全相同的日期重复出现，您的问题就变成一个非常简单的cumsum和cumcount问题:

m = df.Date.eq(df.at[df.index[0], 'Date']).cumsum()
df['RowNum'] = df.groupby(m).cumcount() + 1

<小时/>

如果没有，您可以检查差异:

df['Date'] = pd.to_datetime(df['Date'], errors='coerce')
m = df['Date'].diff().dt.total_seconds().fillna(-1).lt(0).cumsum()
df['RowNum'] = df.groupby(m).cumcount() + 1

或者，类似地，通过将​​底层 NumPy 数组转换为 float ，然后比较:

s = pd.Series(df['Date'].values.astype(float), index=df.index)
df['RowNum'] = df.groupby(s.fillna(-1).lt(0).cumsum()).cumcount() + 1