Выполнение функции выполняется только для определенных условий в python
G'day!
Таким образом, у меня есть функция, которая берет элементы из двух списков, первая из которых находится в стандартном формате списка, второй – список списков, внутренние списки, содержащие элементы в виде 3-х кортежей. Мой вывод – это новый список в формате второго списка, содержащий такое же количество элементов в одном и том же количестве внутренних списков, причем некоторые из значений слегка скорректированы в результате прохождения через функцию.
- Учитывая список имен переменных в Python, как мне создать словарь с именами переменных в виде ключей (к значениям переменных)?
- Как переопределить функциональность среза списка в его производном классе
- Как я могу умножить все элементы в списке вместе с Python?
- Python - как отсортировать список числовых значений в порядке возрастания
- Сортировка словаря по списку списков в порядке возрастания
Вот пример кода и примерная функция, где цепочка импортируется из itertools. сначала есть список, например [0,1,2,3,1,5,6,7,1,2,3,5,1,1,2,3,5,6] а второй – некоторый список, такой как [[(13,12,32),(11,444,25)],[(312,443,12),(123,4,123)],[(545,541,1),(561,112,560)]]
def add(x, y): return x + y foo = [add(x, y) for x, y in zip(first, chain(*(chain(*second))))] bar = [foo[i:i+3] for i in range(0, len(foo), 3)] second = [bar[i:i+2] for i in range(0, len(foo) / 3, 2)]
* * Примечание. Часть Chain (chain ()) предназначена для следующей цели: потому что в целом сложнее обрабатывать список списка, содержащий 3-х кортежей. Цепочка (chain ()) просто сглаживается (в традиционный список отдельных элементов), второй список с вышеупомянутым «нечетным форматом». Остальная часть кода просто перестраивает новый список в исходный формат с выхода функции, которая уже находится в сплющенной форме.
Проблемы, которые у меня возникают, таковы:
Я хочу, чтобы результат был того же размера и формата, что и исходный «второй» список. Если оба списка пустые, я хочу, чтобы пустой список возвращался. Если первый список пуст, я хочу вернуть исходный второй список. Если второй список пуст, я хочу вернуть пустой список.
Если первый список короче второго списка, я хочу, чтобы функция выполнялась, но элементы могут быть сопоставлены между двумя списками, а затем «избыток» второго списка остается неизменным.
Если второй список короче первого списка, я хочу, чтобы функция выполнялась для того, чтобы многие элементы находились во втором списке, а затем просто игнорировали «лишние» элементы из списка 1, тем самым выводя новый список, который имеет тот же самый размеры и форматирование в качестве исходного второго списка.
Моя проблема в том, что я понятия не имею, как реализовать эти маленькие нюансы в моем коде. Любая помощь будет оценена по достоинству.
Привет, Джеймс
- Вложенные списки и списки
- Самый эффективный способ поиска в списке dicts
- Повторное выражение поиска Python из переменной внутри списка
- эффективно знать, если пересечение двух списков пусто или нет, в python
- Python перебирает список и возвращает значения «вне последовательности»
3 Solutions collect form web for “Выполнение функции выполняется только для определенных условий в python”
Не могли бы вы поместить первый список с None, где его недостаточно долго, и обрезать его там, где он слишком длинный.
то выполняем только функцию, где x не является None, иначе верните y
я попытался закодировать пример
from itertools import chain first = [0, 1, 2, 3, 1, 5, 6, 7, 1, 2, 3, 5, 1, 1, 2, 3, 5, 6] second = [ [(13, 12, 32), (11, 444, 25)], [(312, 443, 12), (123, 4, 123)], [(545, 541, 1), (561, 112, 560)], [(13, 12, 32), (11, 444, 25)], [(312, 443, 12), (123, 4, 123)], [(545, 541, 1), (561, 112, 560)], ] def add(x, y): return x + y def pad(list,length): for i in range(length-len(list)): list.append(None) return list[0:length] first = pad(first,len(list(chain(*(chain(*second))) ))) # There is probably a better way to achieve this foo = [add(x, y) if x else y for x, y in zip(first, chain(*(chain(*second))))] bar = [foo[i:i+3] for i in range(0, len(foo), 3)] second = [bar[i:i+2] for i in range(0, len(foo) / 3, 2)] print second
поскольку zip только зацикливается на меньшем из двух списков, здесь он не слишком полезен. Вы можете создать свой собственный алгоритм, который применяет функцию к двум спискам так, как вы указываете:
from itertools import * def flatten(seq): return list(chain(*(chain(*seq)))) def special_apply(a,b, func): """ applies a two argument function to the given flat lists. The result will have the same size as the second list, even if the first list is shorter. """ result = [] for i in range(len(b)): if i < len(a): element = func(a[i], b[i]) #if a ran out of elements, just supply an unmodified element of b else: element = b[i] result.append(element) return result def add(x,y): return x+y a = [1,1,1] b = [[(13,12,32),(11,444,25)],[(312,443,12),(123,4,123)],[(545,541,1),(561,112,560)]] foo = special_apply(a, flatten(b), add) bar = [foo[i:i+3] for i in range(0, len(foo), 3)] result = [bar[i:i+2] for i in range(0, len(foo) / 3, 2)] print result
Результат:
[[[14, 13, 33], [11, 444, 25]], [[312, 443, 12], [123, 4, 123]], [[545, 541, 1], [561, 112, 560]]]
Я думаю, что это делает все, что вы хотите, если я понял все требования. Основное отличие от вашего кода заключается в том, что он также использует izip_longest() из itertools с пользовательским значением fillvalue , а не простой zip() . Он также просто проверяет специальные случаи, связанные с пустым списком входных данных в начале, что казалось проще, чем пытаться разработать методы составления списка или что-то еще для их обработки.
from itertools import chain, izip_longest def add(x, y): return x + y def func(first, second): if not first: return second if not second: return [] second = chain(*(chain(*second))) # flatten foo = [add(x, y) for x, y in izip_longest(first, second, fillvalue=0)] bar = [tuple(foo[i:i+3]) for i in range(0, len(foo), 3)] return [bar[i:i+2] for i in range(0, len(foo) / 3, 2)] if __name__ == '__main__': first = [ 0, 1, 2, 3, 1, 5, 6, 7, 1, 2, 3, 5, 1, 1, 2, 3, 5, 6] second = [ [(13, 12, 32), (11, 444, 25)], [(312, 443, 12), (123, 4, 123)], [(545, 541, 1), (561, 112, 560)], [(13, 12, 32), (11, 444, 25)], [(312, 443, 12), (123, 4, 123)], [(545, 541, 1), (561, 112, 560)], ] print func(first, second) print print func(first[:-1], second) # 1st shorter, as many as poss, rest unchanged print print func(first, second[:-1]) # 2nd shorter, do only as many as in second
- Как заменить номера с порядком в списке (python)
- Как изменить список с помощью рекурсии в Python?
- Переопределение метода добавления после наследования из списка Python
- Как отсортировать список в зависимости от значений в подсписках?
- Разделение на последний разделитель в строке Python?
- Python: слияние двух списков словарей
- Объединение байтового списка с помощью python
- Добавить в список в словаре
- Как сделать копию списка объектов, которые не изменяются при изменении исходного списка?
- Распаковать список в Python?
- Python проверяет, находятся ли элементы в списке
- Создана таблица из списка, не отображающая все элементы