Как заархивировать элементы в 2 списках только при выполнении условия - python

Question

Когда у меня есть два списка:

list1 = ['course1', 'course2', 'course3', 'course4', 'course5']
list2 = [12, 3, 3, 4, 5]

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

Answer 1

Я предлагаю более короткую версию, используя понимание списка:

[x for x in zip(list1, list2) if x[1] >= 5.5]

Answer 2

Если вы хотите получить сжатый список списков из list1 и list2 с условием, что элемент list2 должен быть >=5.5, попробуйте следующее:

list3 = [[i,j] for i,j in zip(list1, list2) if j>=5.5]

ВЫХОД :

[['course1', 12]]

Answer 3

молнии с map():

list1 = ['course1', 'course2', 'course3', 'course4', 'course5']
list2 = [12, 3, 3, 4, 5]

m = map(lambda x, y: (x, y) if y > 5.5 else None, list1, list2)
list(filter(None.__ne__, m))
# [('course1', 12)]

Answer 4

Забудьте zip и создайте собственное решение:

list1 = ['course1', 'course2', 'course3', 'course4', 'course5']
list2 = [12, 3, 3, 4, 5]


def zip_condition(list_a, list_b, func):
    iter_a = iter(list_a)
    iter_b = iter(list_b)
    while 1:
        try:
            a, b = next(iter_a), next(iter_b)
        except StopIteration:
            return
        if func((a, b)):
            yield (a, b)


result = list(zip_condition(list1, list2, lambda x: x[1] >= 5.5))
print(result)