当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python numpy.putmask()用法及代码示例


借助numpy.putmask()方法,我们可以在条件和给定值的帮助下,通过使用来更改数组中的元素numpy.putmask()方法。

用法: numpy.putmask(array, condition, value)
返回:Return the array having new elements according to value.

范例1:
在这个例子中,我们可以通过使用numpy.putmask()方法,我们可以在给定条件和值的帮助下获取新数组。


# import numpy 
import numpy as np 
  
# using numpy.putmask() method 
arr = np.array([1, 2, 3, 4, 5, 6]) 
np.putmask(arr, arr % 2 == 0, 0) 
  
print(arr)

输出:

array([1, 0, 3, 0, 5, 0])

范例2:

# import numpy 
import numpy as np 
  
# using numpy.putmask() method 
arr = np.array([[1, 2, 3], 
                [3, 2, 1], 
                [1, 2, 3]]) 
  
np.putmask(arr, arr>2, 4) 
  
print(arr)

输出:

[[1 2 4]
[4 2 1]
[1 2 4]]



相关用法


注:本文由纯净天空筛选整理自Jitender_1998大神的英文原创作品 Python | numpy.putmask() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。