import numpy as np
a=np.array(list(range(3*4*5))).reshape((3,4,5))
print('a=',a)
print('-'*36+'>')
b=a[...,0]>5
print('b=',b)
print('-'*36+'>')
print('a[0,b[0]]=',a[0,b[0]])
print('-'*36+'>')
a= [[[ 0 1 2 3 4]
[ 5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]]
[[20 21 22 23 24]
[25 26 27 28 29]
[30 31 32 33 34]
[35 36 37 38 39]]
[[40 41 42 43 44]
[45 46 47 48 49]
[50 51 52 53 54]
[55 56 57 58 59]]]
------------------------------------>
b= [[False False True True]
[ True True True True]
[ True True True True]]
------------------------------------>
a[0,b[0]]= [[10 11 12 13 14]
[15 16 17 18 19]]
------------------------------------>