Pandas：在 2 个索引值之间随机拆分索引值

Question

这是一个“ISO 第 53 周问题”。

我有一个 pandas Series 实例，其索引值代表 ISO 周数：

import pandas as pd
ts = pd.Series([1,1,1,2,3,1,2], index=[1,1,2,2,52,53,53])

我想用index = 52 或index = 1 随机且均等地替换所有index = 53 索引。

对于上述情况，这可能是：

import pandas as pd
ts = pd.Series([1,1,1,2,3,1,2], index=[1,1,2,2,52,52,1])

或

import pandas as pd
ts = pd.Series([1,1,1,2,3,1,2], index=[1,1,2,2,52,1,52])

例如。 请问我该怎么做？

感谢您的帮助。

编辑

在 numpy 中，我使用以下方法来实现：

from numpy import where
from numpy.random import shuffle

indices = where(timestamps == 53)[0]
number_of_indices = len(indices)
if number_of_indices == 0:
    return # no iso week number 53 to fix.
shuffle(indices) # randomly shuffle the indices.
midway_index = number_of_indices // 2
timestamps[indices[midway_index:]] = 52 # precedence if only 1 timestamp.
timestamps[indices[: midway_index]] = 1

timestamps 数组是 pandas 的 index 值。

Answer 1

如果我理解正确，列表理解应该可以工作：

ts = pd.Series([1,1,1,2,3,1,2], index=[1,1,2,2,52,53,53])
ts.index = [i if i != 53 else np.random.choice([1,52]) for i in ts.index]

1     1
1     1
2     1
2     2
52    3
52    1
1     2
dtype: int64