如何cplot()一个条件函数

我想绘制一个在复数上定义的函数:如果floor(Re(z))%2==0,f(z)=z,否则f(z=sin(z)

我尝试编写以下代码:

import cplot
import math
import numpy as np

def f(z):
    if math.floor(np.real(z)[0])%2==0:
        res =z
    else:
        res=np.sin(z)  
    return res

plt = cplot.plot(f, (-10, +10, 3000), (-10, 10, 3000),5)
plt.show()

但它不起作用,程序完全忽略了罪恶函数的存在,我不知道为什么。

我认为这是因为我使用了 math. 和 np. 所以我测试了一个更简单的函数:

如果Re(z)>=0,f(z)=z,否则f(z”=sin(z)

import cplot
import numpy as np

def f(z):
    if np.real(z)[0]>=0:
        res =z
    else:
        res=np.sin(z)  
    return res

plt = cplot.plot(f, (-10, +10, 3000), (-10, 10, 3000),5)
plt.show()

这一次的存在 z 完全被忽视了。

我想问题可能来自 [0] 在里面 np.real(z)[0] 但删除它会产生错误

ValueError:包含多个元素的数组的真值不明确。使用a.any()或a.all()

我用了它们,结果还是一样的,它完全忽略了 z 在我的第二段代码中,问题似乎也出在 > 标志;如果我把它改成 < 它完全忽略了 sin .