Numpy从变量进行切片

Question

我正在尝试使用预定义的变量来切片一个numpy数组。这样做是有效的：

b = fromfunction(lambda x,y: 10*x+y, (5,4),dtype=int) # Just some matrix

b[1:3,1:3]
# Output:
# array([[11, 12],
#       [21, 22]])

但是我想做的是这样的：

slice = "1:3,1:3"
b[slice]
# Output:
# array([[11, 12],
#       [21, 22]])

对于我来说，切片变量的类型并不重要，我只是举了一个字符串的例子。我该如何保存这样的切片限定符？

- XerXes

Answer 1

您可以使用内置的slice函数。

s = slice(1,3)
b[s,s]

ds = (s,s)
b[ds]

- Pierre GM

Answer 2

numpy.s_和numpy.index_exp提供了一种方便的方式来实现这一点:

the_slice = numpy.index_exp[1:3, 1:3]
b[the_slice]

使用切片、元组、None 和 Ellipsis 的组合可以实现与它们相同的功能，但是它们允许您使用与数组切片完全相同的语法（s_ 和 index_exp 之间唯一的区别在于对于一维切片，s_ 返回一个切片对象，而 index_exp 将其封装在元组中）。

- James