Как создать в цикле датасеты разного размера и как работать с ними?

Хочу создавать в цикле датасеты с разным количеством столбцов, чтобы можно было обращаться к ним, train_data1, train_data2 и так далее.

for i in range(3):
    # features - здесь создается датасет Pandas с разным количеством признаков 
    train_data = features.iloc[i*100:i*100+500, :] # нужно сделать train_data+i

Как это реализовать?

Или каким образом из всех этих разных train_data[123] создать общую базу, чтобы было удобно обращаться к разным train_data?


Ответы (1 шт):

Автор решения: MaxU

если я правильно понял вопрос:

train_data = {}

for i in range(1, 3+1):
    # features - здесь создается датасет Pandas с разным количеством признаков 
    train_data[f"train{i}"] = features.iloc[i*100:i*100+500, :]

PS f-string (PEP-498) использованыый в ответе был добавлен в версии Python 3.6. Для более старых версий используйте старый стиль форматирования:

train_data["train{}".format(i)] = features.iloc[i*100:i*100+500, :]
→ Ссылка