您的位置：首页 > 编程语言 > Python开发

python如何对数组内的元素进行条件运算?

2017-08-28 10:57 591 查看

python里面数组运算numpy是首选, C的运算效率, 下面是个简单例子

>>import numpy as np
>>arr = np.random.rand(5,5) # 5x5的随机数组
>>arr
array([[ 0.36401504,  0.87729325,  0.53453396,  0.5031851 ,  0.90744627],
[ 0.44607108,  0.22673265,  0.77539262,  0.84064101,  0.41774629],
[ 0.91686641,  0.71480639,  0.09617148,  0.41535823,  0.4175782 ],
[ 0.25753864,  0.73196495,  0.43557823,  0.97497112,  0.54451645],
[ 0.58167148,  0.50382906,  0.88490307,  0.77238085,  0.17427672]])
>>mask = arr>0.5 # 选取数组中元素大于0.5的元素的bool数组
>>mask
array([[False,  True,  True,  True,  True],
[False, False,  True,  True, False],
[ True,  True, False, False, False],
[False,  True, False,  True,  True],
[ True,  True,  True,  True, False]], dtype=bool)
>>arr[mask] = 0 # 将数组中大于0.5的值设为0
>>arr
array([[ 0.36401504,  0.        ,  0.        ,  0.        ,  0.        ],
[ 0.44607108,  0.22673265,  0.        ,  0.        ,  0.41774629],
[ 0.        ,  0.        ,  0.09617148,  0.41535823,  0.4175782 ],
[ 0.25753864,  0.        ,  0.43557823,  0.        ,  0.        ],
[ 0.        ,  0.        ,  0.        ,  0.        ,  0.17427672]])

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： Python 数组条件运算

相关文章推荐

新的分享

章节导航