擅长:python、mysql、java
<pre><code>import numpy as np
xs=np.arange(0,6,1)
def f(a):
it = np.nditer([a, None])
for x, y in it:
y[...] = x if x <= 2 else x * 2
return it.operands[1]
print(f(xs))
</code></pre>
<hr/>
<pre><code>[ 0 1 2 6 8 10]
</code></pre>
<p>抱歉,我没有找到你的错误,但我觉得它可以实现不同的。你知道吗</p>