numpy数组(五)

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]]]
"""
原文地址:https://www.cnblogs.com/peterleee/p/9373757.html