Python объединяет три списка в один в определенном формате
У меня есть три списка, которые я хочу сделать в один список (включая все возможные комбинации 4 для каждой аббревиатуры страны), строка, которая должна быть следующей:
query = f'c_code = {country}&min_age = {age[0]}&max_age = {age[1]}&gender = {gender}'
то есть каждая страна будет иметь 4 строки, я сделал следующие три цикла for, чтобы сделать это, но я считаю, что это не очень pythonic.
country_abb=['ID', 'IR', 'IQ', 'IE', 'IM', 'IL', 'IT', 'CI', 'JM', 'JP', 'JE', 'JO', 'KI', 'KW', 'KG', 'LA', 'LV', 'LB', 'LS', 'LR', 'LY', 'LI', 'LT', 'LU', 'MO', 'MK', 'MG', 'MW', 'MY', 'MV', 'ML', 'MT', 'MH', 'MQ', 'MR', 'MU', 'YT', 'MX', 'FM', 'MD', 'MC', 'MN', 'ME', 'MS', 'NA', 'NP', 'NL', 'NC', 'NZ', 'NI', 'NE', 'NG', 'MP', 'NO', 'OM', 'PK', 'PW']
genders = ['male','female']
ages_range = [[16,25],[26,50]]
all_queries = []
for country in country_abb:
for gender in genders:
for age in ages_range:
query = f'c_code = {country}&min_age = {age[0]}&max_age = {age[1]}&gender = {gender}'
all_queries.append(query)
@roganjosh Исправлено, ЕСЛИ у меня было - вместо запятой
Ответы 2
Вы можете использовать product из itertools. Я не знаю, является ли это более питоническим, но он немного короче, и я нахожу его более читаемым, но я думаю, что ваш пробег может отличаться:
from itertools import product
country_abb=['ID', 'IR', 'IQ', 'IE', 'IM', 'IL', 'IT', 'CI', 'JM', 'JP', 'JE', 'JO', 'KI', 'KW', 'KG', 'LA', 'LV', 'LB', 'LS', 'LR', 'LY', 'LI', 'LT', 'LU', 'MO', 'MK', 'MG', 'MW', 'MY', 'MV', 'ML', 'MT', 'MH', 'MQ', 'MR', 'MU', 'YT', 'MX', 'FM', 'MD', 'MC', 'MN', 'ME', 'MS', 'NA', 'NP', 'NL', 'NC', 'NZ', 'NI', 'NE', 'NG', 'MP', 'NO', 'OM', 'PK', 'PW']
genders = ['male','female']
ages_range = [[16,25],[26,50]]
all_queries = []
for country, gender, age in product(country_abb, genders, ages_range):
query = f'c_code = {country}&min_age = {age[0]}&max_age = {age[1]}&gender = {gender}'
all_queries.append(query)
product создает список списка, в котором объединены все элементы исходных списков, что очень похоже на вложенные циклы for в исходном сообщении. Подробнее об этом можно прочитать в документации Python здесь. (Спасибо, @roganjosh, за предложение добавить ссылку.)
что именно делает продукт?
Он создает список списка, в котором объединены все элементы исходных списков, что очень похоже на вложенные циклы for в исходном сообщении.
даст ли это все комбинации, если я добавлю больше age_ranges, например ages_range = [[16,25],[26,50],[0,15],[51,80]]?
Да, так же, как и с вложенными for петлями.
Вы также можете просто использовать list comprehension:
country_abb=['ID', 'IR', 'IQ', 'IE', 'IM', 'IL', 'IT', 'CI', 'JM', 'JP', 'JE', 'JO', 'KI', 'KW', 'KG', 'LA', 'LV', 'LB', 'LS', 'LR', 'LY', 'LI', 'LT', 'LU', 'MO', 'MK', 'MG', 'MW', 'MY', 'MV', 'ML', 'MT', 'MH', 'MQ', 'MR', 'MU', 'YT', 'MX', 'FM', 'MD', 'MC', 'MN', 'ME', 'MS', 'NA', 'NP', 'NL', 'NC', 'NZ', 'NI', 'NE', 'NG', 'MP', 'NO', 'OM', 'PK', 'PW']
genders = ['male','female']
ages_range = [[16,25],[26,50]]
all_queries = [
f"c_code = {country}&min_age = {age[0]}&max_age = {age[1]}&gender = {gender}"
for age in ages_range
for gender in genders
for country in country_abb
]
хорошо знать, что это можно сделать в одном понимании списка, но в основном это то же самое, что и мой код
@hadesfv да, выглядит лучше, чем вложенный цикл.
@roganjosh дефис в возрастном диапазоне[1]