python - 尝试访问矩阵外部时获取特定值

Question

在尝试获取 NumPy 矩阵之外的值时，我能否收到指定值(例如 0)？

例如:

#getItem - some method which do what i want
a = numpy.array([[1, 2], [3, 4]])
getItem(a, 1, 1) # == 4
getItem(a, -1, 1) # == 0

当然我可以手动完成，但在 Python 中这些方法可能性能不佳。

Answer 1

您可以使用try/except block 来处理无效索引

def getItem(a, i, j):
    try:
        return a[i,j]
    except IndexError:
        return 0

>>> getItem(a, 1, 1)
4
>>> getItem(a, 2, 5)
0