Python / Numpy - Извлечение 2-го подмассива из 2-го массива с использованием переменных

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]

Кажется, работает. Итак, мой вопрос - есть ли способ передать переменные в индекс массива? Если нет, то есть ли лучший способ создать подмассив с использованием переменных?

Я пытался найти эту проблему при переполнении стека, но безуспешно, но я уверен, что я не первый, у кого возникла эта проблема?

Ура, ребята, Morgan

Answer 1

Вы получаете i из цикла, но не используете его.

q = 50

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