Python/Numpy - 使用变量从二维数组中提取二维子数组

Question

好的，所以我有一个二维数据数组，其形状为(23025, 1000)，称为“allfiles”。

我需要一次遍历数组 50 列并将它们提取到子数组中进行操作。问题是当我使用下面的代码寻址数组时，它似乎无法识别变量（a 和 b）。我目前拥有的代码如下所示。

    q = 50
    a = np.shape(allfiles)[1] # a = 1000
    for i in range(a):
        b = a + q
        data = allfiles[:,a:b]

当我用数字替换变量时，即..

    data = allfiles[:,30:80]

它似乎有效。所以，我的问题是 - 有没有办法可以将变量传递到数组索引？如果没有，是否有更好的方法可以使用变量创建子数组？

我试图在堆栈溢出上找到这个问题，但没有成功，但我确信我不是第一个遇到这个麻烦的人？

干杯，伙计们， 摩根

Answer 1

您从循环中获取了 i 但不使用它。

q = 50

for start in xrange(0, allfiles.shape[1], q):
    data = allfiles[:,start:start+q]
    ...