mask = ny.array([1, 0, 1, 0, 0, 1], dtype=ny.bool)
print('a[mask,2]: ', a[mask, 2]) # 另外一种写法,第二列可以输出的行号为true
mask = ny.array([1, 0, 1, 0, 0, 1]) # 不是bool类型就是按下标来取值
print(a[mask, 3])
"""
a[mask,2]: [ 2 22 52]
[13 3 13 3 3 13]
"""
x = ny.array([[0, 1], [2, 3]])
y = ny.array([[-1, -2], [-3, -4]])
print('a[x,y]:
', a[x, y])
"""
a[x,y]:
[[ 5 14]
[23 32]]
"""
print('a[x]:
', a[x]) # 缺省时候用:来代替最后应该是一个三元数组
"""
a[x]:
[[[ 0 1 2 3 4 5]
[10 11 12 13 14 15]]
[[20 21 22 23 24 25]
[30 31 32 33 34 35]]]
"""