Создайте два списка одновременно

Question

Фон

Алгоритм манипулирует финансовой аналитикой. Существует несколько списков одного размера, и они фильтруются в другие списки для анализа. Я делаю одну и ту же фильтрацию по разным параллельным спискам. Я мог бы настроить его так, чтобы a1, b1, c2 появлялись в виде кортежа в списке, но тогда аналитики должны разделить кортежи другим способом анализа (регрессия одного списка против другого, бета и т. Д.).

Что я хочу сделать

Я хочу создать два разных списка на основе третьего списка:

>>> a = list(range(10))
>>> b = list(range(10,20))
>>> c = list(i & 1 for i in range(10))
>>>
>>> aprime = [a1 for a1, c1 in zip(a,c) if c1 == 0]
>>> bprime = [b1 for b1, c1 in zip(b,c) if c1 == 0]
>>> aprime
[0, 2, 4, 6, 8]
>>> bprime
[10, 12, 14, 16, 18]

Кажется, должен быть способ pythonic / функционального программирования / itertools для создания двух списков и повторения этих трех списков только один раз. Что-то вроде:

aprime, bprime = [a1, b1 for a1, b1, c1 in zip(a,b,c) if c1 == 0]

Но, конечно, это приводит к синтаксической ошибке.

Вопрос

Есть ли питонический путь?

Перестрелка микрооптимизации

Ужасный, но питон-до-макс., Однострочник исключает решение "просто используй для петли" и мой оригинальный код в очень популярном timeit совпадении клетки:

>>> import timeit
>>> timeit.timeit("z2(a,b,c)", "n=100;a = list(range(n)); b = list(range(10,10+n)); c = list(i & 1 for i in range(n));\ndef z2(a,b,c):\n\treturn zip(*[(a1,b1) for a1,b1,c1 in zip(a,b,c) if c1==0])\n")
26.977873025761482
>>> timeit.timeit("z2(a,b,c)", "n=100;a = list(range(n)); b = list(range(10,10+n)); c = list(i & 1 for i in range(n));\ndef z2(a,b,c):\n\taprime, bprime = [], [];\n\tfor a1, b1, c1 in zip(a, b, c):\n\t\tif c1 == 0:\n\t\t\taprime.append(a1);  bprime.append(b1);\n\treturn aprime, bprime\n")
32.232914169258947
>>> timeit.timeit("z2(a,b,c)", "n=100;a = list(range(n)); b = list(range(10,10+n)); c = list(i & 1 for i in range(n));\ndef z2(a,b,c):\n\treturn [a1 for a1, c1 in zip(a,c) if c1 == 0], [b1 for b1, c1 in zip(b,c) if c1 == 0]\n")
32.37302275847901

Answer 1

Просто используйте петлю for:

aprime = []
bprime = []
for a1, b1, c1 in zip(a, b, c):
    if c1 == 0:
        aprime.append(a1) 
        bprime.append(b1) 

Answer 2

Это может выиграть самую уродливую награду за код, но работает в одной строке:

aprime, bprime = zip(*[(a1,b1) for a1,b1,c1 in zip(a,b,c) if c1==0])

Answer 3

Невозможно создать несколько списков одновременно с использованием списков - если вы хотите выполнять итерацию только тогда, когда вам понадобится сделать это другим способом - возможно с циклом.

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