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 是輸入序列的數量。這些陣列共同構成一個開放網格。
範例
>>> 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]])