Как построить 2d функцию условия

Я хочу построить двухмерную функцию условия и код, как показано ниже:

from numpy import exp,arange
from pylab import meshgrid,cm,imshow,show

def z_func(x,y):
    zParity = x % 2
    nParity = y % 2
    if zParity == 0 and nParity == 0:
        return x+y+1
    elif zParity!=0 and nParity!=0:
        return x+y-1
    else:
        return x+y

x = arange(1,100,1)
y = arange(1,100,1)
X,Y = meshgrid(x, y) # grid of point
Z = z_func(X, Y) # evaluation of the function on the grid
im = imshow(Z,cmap=cm.RdBu) # drawing the function
show()

Я получаю сообщение об ошибке: ValueError: значение истинности массива с более чем одним элементом неоднозначно. Используйте a.any() или a.all() Как я могу изменить код, чтобы он работал?

python

14.05.2022 17:01

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Попробуй это:

from numpy import exp,arange
from pylab import meshgrid,cm,imshow,show
import numpy as np

def z_func(x,y):
    zParity = x % 2
    nParity = y % 2
    if np.all((zParity == 0, nParity == 0)):
        return x+y+1
    elif np.all((zParity!=0, nParity!=0)):
        return x+y-1
    else:
        return x+y

x = arange(1,100,1)
y = arange(1,100,1)
X,Y = meshgrid(x, y) # grid of point
Z = z_func(X, Y) # evaluation of the function on the grid
im = imshow(Z,cmap=cm.RdBu) # drawing the function
show()

Вот что я получаю после запуска кода (надеюсь, я понял, что вы хотите получить, иначе я соответствующим образом отредактирую свой ответ):