У меня есть список Python, например list = [1,2,3,4,5,6,7,8,9]
. Кроме того, у меня есть еще один список list2 = ['a','b','c','d','e','f','g','h','j']
. Теперь я хотел бы сделать следующее:
idx = (list > 3) & (list < 7)
list2 = list2[idx]
Это должно дать ['d','e','f']
. Конечно, это невозможно со списками. Как это сделать со списками?
Вы можете использовать zip
:
l1 = [1,2,3,4,5,6,7,8,9]
l2 = ['a','b','c','d','e','f','g','h','j']
result = [a for a, b in zip(l2, l1) if 3 < b < 7]
Выход:
['d', 'e', 'f']
Чтобы также получить сокращенный список:
result, reduced = map(list, zip(*[[a, b] for a, b in zip(l2, l1) if 3 < b < 7]))
Выход:
['d', 'e', 'f'] #result
[4, 5, 6] #reduced
Это работает. Как я могу также получить сокращенный список l1? Это означает [4,5,6].
Попробуй это:
#Make sure all lists are numpy arrays.
import numpy as np
idx = np.where((list > 3) & (list < 7))
list2 = list2[idx]
Вариант со списком доступа к элементам по индексу:
res = [ list2[idx] for idx, e in enumerate(list1) if 7 > e > 3 ]
print(res) #=> ['d', 'e', 'f']
[ [list2[idx], e] for idx, e in enumerate(list1) if 7 > e > 3 ]
#=> [['d', 4], ['e', 5], ['f', 6]]
Затем транспонируйте:
res1, res2 = [ list(i) for i in zip(*[ [list2[idx], e] for idx, e in enumerate(list1) if 7 > e > 3 ]) ]
print (res1, res2 ) #=> ['d', 'e', 'f'] [4, 5, 6]
Также проверьте
enumerate