是否有一种有效的方法来传递“全部”作为数字索引？

Question

我有代码生成一个布尔数组，该阵列用作numpy阵列上的掩码，沿着：

def func():
    a    = numpy.arange(10)
    mask = a % 2 == 0
    return a[mask]

---

现在，我需要将其分为创建掩码的情况，并且未创建它并且使用了所有值反而。这可以实现如下：

def func(use_mask):
    a    = numpy.arange(10)
    if use_mask:
        mask = a % 2 == 0
    else:
        mask = numpy.ones(10, dtype=bool)
    return a[mask]

但是，对于大型阵列而言，这变得极为浪费，因为必须首先创建同样大的布尔阵列。

因此，我的问题是：是否可以将某些内容作为“索引”来重现此类无处不在的数组的行为

？对于涉及一些索引魔术等的其他内容是一个有效的解决方案，但是只需通过扩展的案例区别避免掩盖掩蔽，或者不需要更改代码结构的其他内容，因为它会损害可读性和可维护性（请参阅下一段）。

---

为了完整的目的，这是我目前正在考虑的事情，尽管这会使代码变得更加混乱，而且更简化，因为它扩展了IF/elly在技术上需要的位置（实际上，实际使用了一次掩码，不止一次使用，因此，每个发生的情况都需要在情况区分中包含；

def func(use_mask):
    a    = numpy.arange(10)
    if use_mask:
        mask = a % 2 == 0
        r   = f1(a[mask])
        q   = f2(a[mask], r)
        return q
    else:
        r   = f1(a)
        q   = f2(a, r)
        return q

Answer 1

回想一下a [：]返回a的内容（即使a是多维的，即使是多维）。我们不能将：存储在mask变量中，但是我们可以等效地使用slice对象：

def func(use_mask):
    a = numpy.arange(10)
    if use_mask:
        mask = a % 2 == 0
    else:
        mask = slice(None)
    return a[mask]

这不使用任何内存来创建索引大批。我不确定a [slice（none）]操作的CPU用法是什么。