numpy 中是否存在二維“位置”？

Question

這似乎是一個奇怪的問題，但它歸結為一個非常簡單的操作，我找不到 numpy 的等效項。 我查看了np.where以及許多其他操作，但找不到執行此操作的任何內容：

a = np.array([1,2,3])
b = np.array([1,2,3,4])
c = np.array([i<b for i in a])

output 是一個二維數組 (3,4)，由比較每個值的布爾值組成。

Answer 1

如果你問如何在沒有循環的情況下獲得c ，試試這個

# make "a" a column vector
# > broadcasts to produce a len(a) x len(b) array
c = b > a[:, None]
c
array([[False,  True,  True,  True],
       [False, False,  True,  True],
       [False, False, False,  True]])

Answer 2

您可以擴展其他答案中的方法以獲取a和b的值。 給定一個面具

c = b > a[:, None]

您可以使用np.where或np.nonzero提取每個維度的索引：

row, col = np.nonzero(c)

並使用索引獲取相應的值：

ag = a[row]
bg = b[col]

結果中可能會重復a和b的元素。