Является ли назначение двух переменных элементам в одном списке лучшим способом доступа к этим элементам и выполнения операций с ними?
Скажем, у меня есть список
l = [1,2,3]
и я хочу суммировать каждый элемент в списке с каждым другим элементом в списке. Я мог бы сделать это:
x = [(a, b) for a in l for b in l]
y = [(a + b) for a in l for b in l]
x = [(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3), (3, 1), (3, 2), (3, 3)]
y = [2, 3, 4, 3, 4, 5, 4, 5, 6]
Является ли назначение двух переменных (a и b) элементам в одном списке (l) лучшим способом доступа и выполнения операций с этими элементами? Есть лучший способ сделать это?
Я просмотрел список методов и функций и не смог их найти.
ОБНОВИТЬ
Многие люди рекомендовали функцию продукта itertools, поэтому я подумал, что включу время, прежде чем пометить как ответ:
mysetup = """
from itertools import product
l = [10, 15, 3, 7]
k = 17
"""
mycode = """
[i for i in product(l, repeat=2)]
"""
t = timeit.timeit(setup=mysetup, stmt=mycode, number=100000)
# 0.1505305
mysetup2 = """
l = [10, 15, 3, 7]
k = 17
"""
mycode2 = """
[(a, b) for a in l for b in l]
"""
t1 = timeit.timeit(setup=mysetup2, stmt=mycode2, number=100000)
# 0.1432976
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
6
0
102
4
Перейти к ответу
Данный вопрос помечен как решенный
Ответы 4
Попробуйте использовать itertools.product, который получает каждую комбинацию последовательности:
>>> from itertools import product
>>> l = [1,2,3]
>>> list(product(l, repeat=2))
[(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3), (3, 1), (3, 2), (3, 3)]
>>> list(map(sum, product(l, repeat=2)))
[2, 3, 4, 3, 4, 5, 4, 5, 6]
>>>
Попробуйте это, itertools.продукт
>>> from itertools import product
>>> l = [1,2,3]
>>> [sum(combination) for combination in list(product(l, repeat = 2))]
[2, 3, 4, 3, 4, 5, 4, 5, 6]
@ThierryLathuille Спасибо за указание. Ответ был обновлен с помощью product.
—
29.05.2019 12:49
Вот подход, основанный на itertools:
from operator import add
from itertools import product, starmap
l = [1,2,3]
list(starmap(add, product(l, repeat=2)))
# [2, 3, 4, 3, 4, 5, 4, 5, 6]
Давайте проверим тайминги:
l = list(range(1000))
%timeit list(map(sum, product(l, repeat=2)))
# 187 ms ± 14.4 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
%timeit [sum(combination) for combination in combinations_with_replacement(l, 2)]
# 123 ms ± 4.32 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)
list(starmap(add, product(l, repeat=2)))
# 102 ms ± 4.11 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)
combinations_with_replacement не дает те же значения, что и product - он не дает пары в обратном порядке, поэтому он примерно в два раза быстрее...
—
29.05.2019 12:39
Спасибо, что указали на @ThierryLathuille, действительно здесь мы хотим декартово произведение.
—
29.05.2019 12:43
Ответ принят как подходящий
Python предоставляет встроенный метод
from itertools import product
l = [1,2,3]
Затем сгенерируйте сумму, используя понимание списка за один шаг, чтобы быть более эффективным.
result= [sum(i) for i in product(l, repeat= 2) ]
#result=[2, 3, 4, 3, 4, 5, 4, 5, 6]
Другие вопросы по теме
Сплайн в 3D не может быть дифференцирован из-за AttributeError
Как использовать сервер языка python для vscode по определенному пути?
Получите среднее количество записей в месяц с датой и временем в Pandas
Как минимизировать время преобразования numpy ndarray в список
Синтаксический анализатор Python для преобразования текста в XML
Динамически создавать класс, производный от класса ABC
Преобразование некоторых данных базы данных (таблица SQL) в формат .csv
Как просмотреть XML, созданный пакетом python-docx
Разобрать числа в скобках на отрицательные числа
Как начать перечислять ключи после новой строки в блоке
Похожие вопросы
Check_MK не читает локальную проверку Python на удаленном хосте
Линейная регрессия TensorFlow — возврат NaN для весов, смещения и Inf для потерь
Есть ли метод SciPy для автоматической обрезки изображения, т.е. е. обрезать нули из 2d `numpy.ndarray`?
Как получить индекс тома файлов RAR с помощью команд 7-zip (или любых альтернативных решений)
Алгоритм игры двух стеков (проблема рангов хакеров)
Получение данных из формы с помощью GET данных. Джанго
Изменить метку времени TZ в искре
Как исправить: KeyError: '__file__' в sitecustomize.py (anaconda3) при использовании файла запуска (F5)
Возвращает «Ошибка», если совпадение не найдено регулярным выражением
Ошибка при загрузке fmu в python с библиотекой pyfmi
combinations_with_replacementне дает тех значений, которые хотел ОП. Вместо этого вы должны использоватьproduct, но синхронизация показывает, что он медленнее, чем код в вопросе.