Python 使用内置切片函数对二维数组进行切片

我会试着把你的问题分解一下:

我想不出如何为2D数组创建切片对象，因为Slice函数不接受像slice(0, :)这样的冒号运算符

实际上，这是行不通的，因为slice对象不是这样工作的:它们表示沿one dimension的索引范围，而您想要沿two dimensions索引.因此，虽然这不适用于单个slice对象，但您可以使用元组-比较文档here，特别是第Slicing and striding节的最后一个项目符号:

切片元组始终可以构造为obj，并在x[obj]表示法中使用.切片对象可以在构造中用来代替[start:stop:step]表示法.例如，x[1:10:5, ::-1]也可以实现为obj = (slice(1, 10, 5), slice(None, None, -1)); x[obj].

在您的情况下，因为您只要求完整的行和列:

我需要编写一个从2D数组返回单行或单列的函数.

…相应的:可以写成slice(None, None, None).事实上，您甚至不需要显式地创建切片元组，因为您只有两个索引(一个用于行，一个用于列，它们都是:)，您可以直接使用它们来索引array.因此，如果您真的想将索引打包到一个函数中，我将创建一个函数，如下所示:

def get_slice(row=slice(None,None,None), col=slice(None,None,None)):
    return a[row, col]  # or, equivalently a[(row, col)]

在这里，我们将提供:或等价的slice(None,None,None)作为行和列的缺省值，这样用户只需要为他们真正感兴趣的索引传递一个值.当然，您也可以调整函数，以便接受元组，这将更接近您的原始代码.

在任何情况下，上述版本都将为您提供:

print(get_slice(row=0))  # First row
# >>> [1 2 3]
print(get_slice(col=-1))  # Last column
# >>> [3 6 9]
print(get_slice(col=0))  # First column
# >>> [1 4 7]

最后:

另外，有没有一种方法来表示"最后一行"或"最后一列"…？

您可以使用negative indexing来实现这一点，我已经在上面的示例中的最后一个专栏中使用了它.负索引的工作方式是"从右开始"或"从下开始"，因此a[-1]指的是最后一行，a[:, -2]指的是倒数第二列，依此类推，不需要知道数组的实际形状.