对np.where()函数做的一些测试

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
b = np.array([[0,2,1],[1,2,-1],[0,0,1]])  # 只要非零数，无论多大，其逻辑值都是真
c = b.flatten()
x = np.where(b)  # 此时where返回的是b中逻辑值为真的位置的索引序列
y = np.where(b > 1)  # 条件运算返回的是一个“真值表”。
z = np.where(b, 't', 'f')  # 此时where的第一个参数相当于我们给它输入了一个true、false逻辑序列，它按遍历这个序列把true项换为't',false项换为'f'。

print('x=',x)
print('y=',y)
print('z=',z)
print('条件运算测试1：', a % 2 == 0)
print('条件运算测试2：', b > 0)
print('条件运算测试3：', b > 1)

m = np.where(a > 2, '大', '小')
print('验证：', m)

'''
结论：
0、数组进行条件运算返回逻辑值数组（广播条件运算）；
1、where函数输入的数组永远会被当做逻辑值数组来对待，只是随着参数数量不同而产生不同形式的返回值；
2、如果where里只有1个参数（输入的逻辑值数组），那么返回值就是这个数组中其逻辑值为真的位置的索引；
3、如果where里有3个参数（数组，真值替换项，假值替换项），那么返回值就是原数组相应位置被替换后的同形状的新数组。
'''

输出：