我希望能够在2D数组中的特定元素上执行操作,如果它符合标准的话。在下面的示例中,代码生成任何值<;0.5 = 0.
有人知道这样做的简单方法吗?下面是我的代码,但我相信有一个更简单的方法
import numpy as np
x = 5
y = 5
x3 = np.random.rand(x,y)
def choice(arr):
row = -1
column = -1
for i in arr:
row += 1
for j in i:
column += 1
if j >= 0.5:
arr[row,column] = j
else:
arr[row,column] = 0
if column == y - 1:
column = -1
return arr
y3 = choice(x3.copy())
将所有<;0.5
仅针对枚举和三元条件介绍,您可以执行以下操作:
但AKX解决方案更好
相关问题 更多 >
编程相关推荐