目录
示例
import numpy as np
a = np.arange(20).reshape(4, 5)
"""
[[ 0 1 2 3 4]
[ 5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]]
"""
b1 = [False, True, False, False] # [1],等同于:[1, 1, 1]
b2 = [False, True, True, False, True] # [1, 2, 4]
print(a[b1, b2]) # [6 7 9]
上次说到要先取b1、b2中True的位置,形成整数数组索引。
这里麻烦了,b1只能一个True,没法对应啊。
此时不报错,会触发NumPy广播机制,即把[1]变成[1 1 1]。