Можно ли сократить код?

Question

[MazAlVlad]

Имеем
входные данные: последовательности a,b,c
имеем последовательность значений VALUE
к этим последовательностям применяется функция any_func
Смысл за один перебор последовательности VALUE обработать данные,
сам по себе VALUE по факту бесконечен и его прекращает if (End_a[0]==True and...

def any_func (End,Value,X,Count):
    Count=Count+1
    if Value>X:
        End=True
    return End,Count

def My_Code(a,b,c):

    #End flags
    End_a=[False,False,False,False,False]
    End_b=[False,False,False,False,False]
    End_c=[False,False,False,False,False]
    #End_d=...
    #End_e=...
    #...

    #Count result,output data
    Count_a=[0,0,0,0,0]
    Count_b=[0,0,0,0,0]
    Count_c=[0,0,0,0,0]
    #Count_d=...
    #Count_e=...
    #...

    #VALUE - any data to calculation
    for VALUE in range(100):            
        
        #indexes
        for j in range(5):              
            if End_a[j]==False:
                End_a[j],Count_a[j]=any_func(End_a[j],VALUE,a[j],Count_a[j])
            if End_b[j]==False:
                End_b[j],Count_b[j]=any_func(End_b[j],VALUE,b[j],Count_b[j])
            if End_c[j]==False:
                End_c[j],Count_c[j]=any_func(End_c[j],VALUE,c[j],Count_c[j])
            #if End_d==
                #...
            #if End_e==
                #...
            #...
               
        #stop data if All End = True
        if (End_a[0]==True and End_a[1]==True and End_a[2]==True and End_a[3]==True and End_a[4]==True and
            End_b[0]==True and End_b[1]==True and End_b[2]==True and End_b[3]==True and End_b[4]==True and
            End_c[0]==True and End_c[1]==True and End_c[2]==True and End_c[3]==True and End_c[4]==True):
            break
        
            
    return Count_a,Count_b,Count_c

if __name__=='__main__':
    
    #input data
    a=[1,2,3,4,5]
    b=[2,4,8,10,12]
    c=[7,6,5,4,3]
    #d=...
    #e=...
    #...
    
    Count_a,Count_b,Count_c = My_Code(a,b,c)
    #result
    print(Count_a,Count_b,Count_c)

Можно ли сократить код или упростить, чувствую начали появляться d,e,f и ...

Answer 1

[Влад Григорьев]

для начала ==True или ==False излишние проверки
потом есть функции any и all которыми можно заменить проверку if (End_a[0]==True ...

#End flags
    End_a=

- все это можно заменить на словарь листов или совсем убрать этот список, а просто считать количество обработок, и сравнивать их с длиной списка.
Также можно заменить списки count_a...

И в целом кажется не нужен тут двойной цикл.
Вы бы лучше нормально задание написали, что за чем и почем надо сделать

Comments

[MazAlVlad]

В принципе это и есть задание.
Должны войти a,b,c и выйти Count_a,Count_b,Count_c.
Тут не двойной цикл, второй цикл просто отправляет индексы в функцию. Первый цикл только для того чтоб не сломать код, по факту там не
for VALUE in range (100):
, а
for VALUE in (список значений): #всегда разной длины и не всегда имеет конец

End_a[],End_b[],End_c[] это даёт понять, что дальше считать a[],b[],c[] не нужно.

Спасибо за совет, почитаю за any и all которыми можно заменить проверку if (End_a[0]==True ...

[MazAlVlad]

"что за чем и почем надо сделать" так для ознакомления
VALUE массив данных торговых сделок.
a,b,c массивы настоящих цен
Count_a,Count_b,Count_c расcчитанные объёмы для a,b,c
any_func() одна из функций для расчёта Count_a,Count_b,Count_c

Т.е. смысл всей этой ерунды VALUE [1,2,3... я не могу отправить в фунцию,
во-первых не знаю какого количества VALUE хватит.
А во-вторых хочу сделать одним циклом для всех a,b,c (скорость)

[Влад Григорьев]

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

Count_a=[0,0,0,0,0]
Count_b=[0,0,0,0,0]
Count_c=[0,0,0,0,0]

counts = [Count_a, Count_b, Count_c]
for els in zip(*counts):
...     print(els)

[Влад Григорьев]

Можно завести какой-то простой но более логичный класс для данных тип и обрабатывать не тупа числа а какой-то логичный объект:

>>> from dataclasses import dataclass
>>> @dataclass
... class El:
...     value: int
...     check: bool = False
...

Answer 2

[Art005]

for j in range(5):              
            if End_a[j]:
                End_a[j],Count_a[j]=any_func(End_a[j],VALUE,a[j],Count_a[j])
if End_a[0] and End_a[1]...

Как минимум это

Answer 3

[o5a]

Не до конца понятна необходимость таких манипуляций. Если это для того, чтобы как-бы 1 раз пройти по VALUES, то по факту это же просто переставляет местами 2 перебора.

перебор по VALUES:
    перебор по всем параметрам:

вместо

перебор по по всем параметрам:
    перебор VALUES:

Насколько я вижу, эта функция просто высчитывает некий порог срабатывания для каждого параметра. Тогда эти функции можно переписать так:

def threshold (values, x):
    for i, val in enumerate(values, 1):
        if val > x:
            return i
    return None

def My_Code(a,b,c):

    #VALUE - any data to calculation
    VALUES = list(range(100))

    Count_a = [threshold(VALUES, x) for x in a]
    Count_b = [threshold(VALUES, x) for x in b]
    Count_c = [threshold(VALUES, x) for x in c]

    return Count_a,Count_b,Count_c

if __name__=='__main__':

    #input data
    a=[1,2,3,4,5]
    b=[2,4,8,10,12]
    c=[7,6,5,4,3]
    #d=...
    #e=...
    #...

    Count_a,Count_b,Count_c = My_Code(a,b,c)
    #result
    print(Count_a,Count_b,Count_c)

# [3, 4, 5, 6, 7] [4, 6, 10, 12, 14] [9, 8, 7, 6, 5]

Не стал менять структуру возвращаемого Counter_a, ..., но вообще это тоже можно объединить во вложенный список, чтобы не было этих отдельных a, b, c, Counter_a, Counter_b, Counter_c, но это уже как удобнее.

В целом если нужно повысить производительность, то лучше посмотреть в сторону numpy, тем более тут чисто работа с массивами чисел.

Comments

[MazAlVlad]

Всё верно. Я то же так раньше сделал, но тут получается не за один перебор VALUES. Тут получается весь VALUES входит в каждую функцию, напрашивается несколько процессов, чтоб не потерять время. И немного не понятно, как потом процессу добавить VALUES, если значений не хватило, значит Process отпадает. Я пытаюсь один VALUES по очереди использовать для всех значений всех функций. Так мне проще добавить значения VALUES, если их не хватило для расчёта Count_a ,Count_b, Count_c... Всё как-то со временем выросло в огромный снежный ком, когда any_func стало несколько: any_func_1; any_func_2; any_func_3... Вроде и код явно повторяется и не соображу, как от этого "леса дров" уйти.

[o5a]

MazAlVlad, но по факту у Вас перебор получается тот же самый, от перестановки мест итераторов кол-во итераций не меняется.

[MazAlVlad]

Фактически верно, что-то мысли в последнее время находят что-то сложное.
Благодарю. На ровном месте трудности создаю, просто генератор трудностей :)
Но тут ещё помозговать нужно. Допустим VALUES=[0,1,2,3] 'это данные за сутки и их явно не хватило, нужно добавить данных за следующие сутки VALUES={4,5,6...14]

#any function
def any_func (End,Value,X,Count):
    Count=Count+1
    if Value>X:
        End=True
    return End,Count

def My_Code(a,b,c):

    #End flags
    End_a=[False,False,False,False,False]
    End_b=[False,False,False,False,False]
    End_c=[False,False,False,False,False]
    #End_d=...
    #End_e=...
    #...

    #Count result,output data
    Count_a=[0,0,0,0,0]
    Count_b=[0,0,0,0,0]
    Count_c=[0,0,0,0,0]
    #Count_d=...
    #Count_e=...
    #...
    
    #запрос данных эти сутки
    DATA=[0,1,2]       
    while True:
        for VALUE in DATA:
            #indexes
            for j in range(5):              
                if End_a[j]==False:
                    End_a[j],Count_a[j]=any_func(End_a[j],VALUE,a[j],Count_a[j])
                if End_b[j]==False:
                    End_b[j],Count_b[j]=any_func(End_b[j],VALUE,b[j],Count_b[j])
                if End_c[j]==False:
                    End_c[j],Count_c[j]=any_func(End_c[j],VALUE,c[j],Count_c[j])
                #if End_d==
                    #...
                #if End_e==
                    #...
                #...
                   
            #stop data if All End = True
            if (all(End_a) and
                all(End_b) and
                all (End_c)):
                print('данных хватило')
                return Count_a,Count_b,Count_c
                
        print('данных не хватило')
        #запрос данных прошлые сутки
        DATA=[3,4,5,6,7,8,9,10,11,12,13,14]
        continue
    
def threshold (values, x):
    for i, val in enumerate(values, 1):
        if val > x:
            return i
    return None

def My_Code2(a,b,c):

    #VALUE - any data to calculation
    VALUES = list(range(14))

    Count_a = [threshold(VALUES, x) for x in a]
    Count_b = [threshold(VALUES, x) for x in b]
    Count_c = [threshold(VALUES, x) for x in c]

    return Count_a,Count_b,Count_c

if __name__=='__main__':
    
    #input data
    a=[1,2,3,4,5]
    b=[2,4,8,10,12]
    c=[7,6,5,4,3]
    #d=...
    #e=...
    #...
    
    Count_a,Count_b,Count_c = My_Code(a,b,c)
    print(Count_a,Count_b,Count_c)
    
    Count_a,Count_b,Count_c = My_Code2(a,b,c)
    print(Count_a,Count_b,Count_c)

Answer 4

[Andrey Dugin]

В код не вникал, но, похоже, вам очень пригодятся эти функции модуля itertools:

from itertools import takewhile, dropwhile, compress