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

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

Answer 2

您可以使用嵌套循环概念。在list中创建一个lists。像：

main_list = list()
for i in range(4):
    temp_list = ["batch{0}".format(i)]
    main_list.append(temp_list)