Python中的转换索引（与R比较）

Question

我检查使用两个系列X和Y，如果它们比另一个大。通过使用loc，我可以获得X>Y为TRUE的系列索引。例如：

X.loc[X>Y]

使用这个索引，我想将索引转换为句点。例如，如果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