python - 有没有办法创建与用户输入一样多的列表？

Question

我想知道如何创建与用户输入一样多的列表。

假设用户输入为 4。

我想创建4个列表名称Batch1 Batch2 Batch3 Batch4

从具有相同 Batch1-4.csv 名称的 csv 文件中检索数据

for i in range(1,3):
    list("Batch{0}".format(i))
    print(Batch1)

我已经尝试过此操作，但导致错误 Batch1 未定义，因为我没有直接声明 Batch1。

你有什么解决办法吗？

Answer 1

您可以像这样使用列表理解:

>>> main_list = ["batch{0}".format(i) for i in range(4)]
>>> main_list
['batch0', 'batch1', 'batch2', 'batch3']

如果您想要列表列表，请执行以下操作:

>>> main_list = [["batch{0}".format(i)] for i in range(4)]
>>> main_list
[['batch0'], ['batch1'], ['batch2'], ['batch3']]

根据用户输入，您的脚本可能如下所示:

n = int(input('Enter a number:'))
main_list = [["batch{0}".format(i)] for i in range(1,n+1)]