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))。

回傳值:
outndarray 的元組

N 個陣列,每個陣列有 N 個維度,其中 N 是輸入序列的數量。這些陣列共同構成一個開放網格。

另請參閱

ogrid, mgrid, meshgrid

範例

>>> import numpy as np
>>> 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]])