在Python中移位索引（與R比較）

Question

如果兩個系列比另一個大，我使用兩個系列X和Y來檢查。通過使用loc，我可以得到我的系列的索引，其中X>Y是TRUE。例如：

X.loc[X>Y] 

使用此索引，我想改變索引n個週期。例如，如果X.loc[X>Y]給我們{1,5,8,9}，我有興趣將它們轉換爲{1+2,5+2,8+2,9+2}。對於這件事我會很感激的！

Answer 1

你可以使用numpy.nonzero拿到指標，然後將它們轉移：

# two random arrays as an example 
X = numpy.random.random(100) 
Y = numpy.random.random(100) 
ids = numpy.nonzero(X > Y)[0] 
print ids 
print ids + 2