Сохранить новую переменную после каждой итерации - PullRequest
0 голосов
/ 27 января 2020

Я новичок в Python и у меня есть вопрос, касающийся помещения функции в функцию для l oop.

Моя проблема: у меня написана функция (см. Ниже), и теперь я хотел бы применить функцию к массиву с 8 строками и 12 столбцами. Мне бы хотелось, чтобы функция обращалась к каждой строке отдельно и возвращала переменную «memlist» для каждой строки массива. Итак, в конце концов, у меня будет memlist_item1, memlist_item2 и т. Д. c. сохранены. Я попытался для l oop с "iterrows", но я не знаю, как продолжить и как обновить имя переменной "memlist" для каждой строки. Может быть, кто-то может помочь. Спасибо!

    def encodelist (w, List1, g, u, t, c):
        memlist = []                               
        ranlist = []
        switchlist = []
        randomlist = np.random.uniform(0.0,1.0,size=List1*w).reshape(List1,w) 

        encodefeatureTF = np.where(randomlist < (1-(1-u)**t),1,0) 

        memlist = np.where(encodefeatureTF, List1, 0)

        ranlist = np.random.uniform(0.0,1.0,size=List1*w).reshape(List1,w)

        switchlist = np.where(((memlist > 0) & (ranlist > c)), 1,0)

        memlist = np.where(switchlist == 0, memlist, np.random.geometric(g, size=1))
        return memlist
...