numpy基础教程--where函数的使⽤
在numpy中，where函数是⼀个三元运算符，函数原型为where(condition, x, y)，意思是当条件成⽴的时候，将矩阵的值设置为x，否则设置为y
⼀个很简单的应⽤就是，在⼀个矩阵当中，将⼤于或等于平均值的数设置为1，将⼩于平均值的数设置为0
1import numpy as np
2# 随机⽣成⼀个3⾏四列的矩阵，范围是1--16
3 np.random.seed(10)
4 t = np.random.randint(1, 16,(3, 4), dtype=int)
5print(t)
6print("*"*30)
7 t_mean = t.mean()
8print("t的平均数为{0}".format(t_mean))
9print("*"*30)
numpy官方教程10# 使⽤where函数可以快速将⼀个矩阵⾥⾯，⼩于平均数的值设置为0，⼤于平均数的值设置为1
11 t1 = np.where(t<t_mean, 0, 1)
12print(t1)
运⾏结果如下图所⽰

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。