python - 有没有一种方法可以创建与用户输入一样多的列表？

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

假设用户输入为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。

您有什么解决方法吗？

最佳答案

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

>>> 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)]