numpy.ix_

numpy.ix_(*args)[source]

Построение открытой сетки из нескольких последовательностей.

Эта функция принимает N одномерных последовательностей и возвращает N результатов с N измерениями каждый, такие, что форма равна 1 во всех измерениях, кроме одного, и измерение с ненулевым значением формы циклически проходит через все N измерений.

Используя ix_ можно быстро построить массивы индексов, которые будут индексировать декартово произведение. a[np.ix_([1,3],[2,5])] возвращает массив [[a[1,2] a[1,5]], [a[3,2] a[3,5]]].

Параметры:

argsОдномерные последовательности

Каждая последовательность должна быть целочисленного или булевого типа. Булевые последовательности будут интерпретироваться как булевые маски для соответствующего измерения (эквивалентно передаче np.nonzero(boolean_sequence)).

Возвращает:

outкортеж массивов ndarrays

N массивов с N измерениями каждый, где N — количество входных последовательностей. Вместе эти массивы образуют открытую сетку.

См. также

ogrid, mgrid, meshgrid

Примеры

>>> a = np.arange(10).reshape(2, 5)
>>> a
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> ixgrid = np.ix_([0, 1], [2, 4])
>>> ixgrid
(array([[0],
       [1]]), array([[2, 4]]))
>>> ixgrid[0].shape, ixgrid[1].shape
((2, 1), (1, 2))
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])

>>> ixgrid = np.ix_([True, True], [2, 4])
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])
>>> ixgrid = np.ix_([True, True], [False, False, True, False, True])
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])