вычитание питона нулевых ранговых массивов

Question

Я пытаюсь вычесть два нумерованных массива numpy a и b. Я бы хотел, чтобы a-b возвращал ошибку, когда они не имеют одинаковый размер. Например, если форма (a) = [300,] и форма (b) = [450,]; a-b возвращает массив формы [300,450]. Вместо этого я хотел бы, чтобы программа возвращала ошибку.

Answer 1

Вы можете попробовать этот подход:

def mySubtract(a, b): 
    if a is None or b is None or a.shape != b.shape: 
     sys.stderr.write("a and b don't have the same shape!\n") 
     raise ValueError("a and b don't have same shape!") 
    return a-b 

Я не был уверен, что вы имели в виду «возвращает ошибку», но если вы предпочитаете просто возвращать неверный результат, а не поднимать исключение ValueError, вместо этого вы можете просто return None или return "a and b don't have the same shape!".

Имейте в виду, что вызывающий абонент вашей программы должен знать, что делать с этими специальными кодами возврата ошибки. Таким образом, в целом исключение может быть более чистым.

Answer 2

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

if a.size() != b.size(): 
    raise Exception("Invalid sizes")