python - 如何在 Pandas 0.24.2 中将 offsets.MonthEnd 系列转换为 int

Question

我计算 pd.Period 类型的两列之间的增量。在 0.24 之前，它将返回一个 int 类型，但在 0.24 中不再返回，它将返回类似一系列的 1 * MonthEnds，2 * MonthEnds，...我想将其转换为 int 类型。

我可以使用 apply 来实现，例如，

df.apply(lambda x: x['z'].n)

或

((df['x'] - df['y']) / np.timedelta64(1, 'M')).round()

但我想知道是否还有其他解决方法。

df = pd.DataFrame({'x':pd.date_range(start='2001-01-01', periods=10), 'y':pd.date_range(start='2002-01-01', periods=10)})

在Pandas 0.24之前，下面的代码会返回一个int类型的列

df['z'] = df['x'].dt.to_period('M') - df['y'].dt.to_period('M')

但是0.24改变了返回类型，如上所述有两种方法仍然返回一个int列，但我想知道是否有其他方法可以实现这一点。

Answer 1

使用 astype 将返回一个 int 而不是 DateOffset 对象:

df['x'].dt.to_period('M').astype(int) - df['y'].dt.to_period('M').astype(int)

0   -12
1   -12
2   -12
3   -12
4   -12
5   -12
6   -12
7   -12
8   -12
9   -12
dtype: int64