在NumPy中使用默认值进行N-D索引

如何解决在NumPy中使用默认值进行N-D索引

我可以对具有边界回退的默认值的NumPy N-D数组进行索引吗？下面是一些虚构的np.get_with_default(a,indexes,default)的示例代码：

import numpy as np
print(np.get_with_default(
    np.array([[1,2,3],[4,5,6]]),# N-D array
    [(np.array([0,1,2]),np.array([1,3,5]))],# N-tuple of indexes along each axis
    13,# Default for out-of-bounds fallback
))

应打印

[2 3 6 13 13 13]

我正在为此寻找一些内置功能。如果不存在，那么至少要执行一些简短有效的实现。

解决方法

我不知道NumPy中是否可以直接进行任何操作，但是您始终可以自己实现。这不是特别聪明或高效，因为它需要多个高级索引操作，但是可以满足您的需求：

----folder1----conditionfolder----conditionfile
  |
  |---folder2----conditionfolder
  |
  |---folder3