python-熊猫:在两个索引值之间随机分割索引值

前端之家收集整理的这篇文章主要介绍了python-熊猫:在两个索引值之间随机分割索引值 前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
@H_301_1@这是“ ISO第53周的问题”.

@H_301_1@我有一个pandas Series实例,其索引值表示ISO周编号:

@H_301_1@

import pandas as pd
ts = pd.Series([1,1,2,3,2],index=[1,52,53,53])
@H_301_1@我想将所有index = 53索引随机且均等地替换为index = 52或index = 1.

@H_301_1@对于上述情况,可能是:

@H_301_1@

import pandas as pd
ts = pd.Series([1,1])
@H_301_1@要么

@H_301_1@

import pandas as pd
ts = pd.Series([1,52])
@H_301_1@例如.请问我该怎么做?

@H_301_1@谢谢你的帮助.

@H_301_1@编辑

@H_301_1@在numpy中,我使用以下方法实现了这一点:

@H_301_1@

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
@H_301_1@其中timestamps数组是熊猫索引值.
最佳答案
如果我对您的理解正确,则列表理解应该可以:

@H_301_1@

ts = pd.Series([1,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
原文链接:https://www.f2er.com/python/533216.html

猜你在找的Python相关文章