tinynumpy icon indicating copy to clipboard operation
tinynumpy copied to clipboard

surpport bool index

Open zhaominxing208 opened this issue 4 years ago • 0 comments

for example: n [112]: arr = np.arange(12).reshape(3,4)

In [113]: i = np.array([True,False,True])

In [114]: i.nonzero() Out[114]: (array([0, 2]),)

In [115]: arr[i] Out[115]: array([[ 0, 1, 2, 3], [ 8, 9, 10, 11]])

In [116]: arr[i.nonzero()] Out[116]: array([[ 0, 1, 2, 3], [ 8, 9, 10, 11]])

zhaominxing208 avatar Jan 07 '22 06:01 zhaominxing208