Numpy 2D индексация с использованием различного среза на столбец [Duplicate] ⇐ Python

[Anonymous]

Код: Выделить всё

import numpy as np

x = np.arange(12).reshape(3, 4)
print(x)

idx = np.array([0, 1])
y = x[:2, idx : idx + 2]

# should be
# [[0 1]
#  [5 6]]
< /code>
Поэтому я хочу получить другой срез для каждой строки. В этом примере это индексы 0, 1 

из первой строки и 1, 2 индексы из второй строки.
Вместо этого я получаю: < /p>

typeerror: только целочисленные скалярные массивы могут быть преобразованы в скалярную индекс < /p>
< /blockquote>
Какие у меня варианты? Есть больше одного? Какие -либо последствия для производительности?
Спасибо

Подробнее здесь: https://stackoverflow.com/questions/793 ... per-column