передавая необязательные аргументы как строки или Arays

Question

У меня есть эта функция, которая работает, как ожидалось

import numpy as np 
def my_fun(X, Weights = None): 
    if Weights is None: 
      np.random.normal(0,1,1) 
    else: 
      Weights + 1 

Поэтому я могу назвать my_fun использованием my_fun (my_X) или my_fun (my_X, Веса = my_weights) где my_weights является массивом.

Однако мне нужно включить третий параметр в my_fun, и в идеале я хотел бы передать аргумент как строку. Что-то вроде этого.

import numpy as np 
    def my_fun(X, Weights = None): 
     if Weights is None: 
       np.random.normal(0,1,1) 
     elif Weights == 'My_string': 
       np.random.beta(1,1,1) 
     else: 
       Weights + 1 

Если я пытаюсь вызвать my_fun в my_fun (my_X, вес = 'а'), то я получаю следующее сообщение об ошибке: FutureWarning: поэлементно сравнение не удалось; возвращающий скаляр вместо ...

Любые мысли о том, как изменить форму my_fun?

Answer 1

Прежде всего, я не вижу цели вашего X, вы никогда не используете его в функции. Зачем тогда посылать?

Также вам нужно return в вашей функции. Что вы хотите вернуть? не ясно. Может быть, что-то похожее.

def my_fun(X, Weights = None): 
    if Weights is None: 
      X = np.random.normal(0,1,1) 
    elif Weights == 'My_string': 
      X = np.random.beta(1,1,1) 
    else: 
      Weights + 1 
    return Weights,X 

print(my_fun([1,2,3],Weights = 'My_string'))