注:下述a为原始多维数组,b为经过替换后的目标多维数组。
方法①:使用numpy的.where()函数
以下语句可以把多维数组a中≥50且≤100的元素都替换为1,否则为0。
b = np.where((a >= 50) & (a <= 100), 1, 0)
参考:
方法②:b[b<=条件值]=目标值
以下语句可以把多维数组a中≤50的元素替换为0,其它元素不变。
b = a.copy()
b[b <= 50] = 0
参考:
P.S.
如果多维数组元素太多而不方便全部查看,可以抽取部分元素查看,进而查看经过元素替换后的多维数组是否满足目标要求。
参考: