时间戳

首页 > 解决方案 > 使用嵌套循环创建嵌套列表

python - 使用嵌套循环创建嵌套列表

问题描述

我想要完成的更大范围是这个。我有一个Windows directory可以包含可变数量的.csv文件。这些文件是作为 PLC 的测试结果生成的。它们通过文件名来区分。每个循环测试生成 2 个.csv文件。第一个文件名包含订单号、-分隔符和字符串UP。第二个文件是相同的，只是字符串被替换为DOWN.

示例文件名：
1234567890-UP或1234567890-DOWN

目录示例：
1234567890-UP.csv
1234567890-DOWN.csv
2000005001-UP.csv
2000005001-DOWN.csv

我正在尝试编写一个循环遍历目录中所有文件名并将它们存储在列表中的脚本。然后从元素中删除除订单号之外的所有内容。然后删除重复的元素。使用上面的示例目录，我将有一个看起来像[1234567890,2000005001]. 我已经完成了这么多。

现在我要做的是遍历文件名的原始列表并将它们与我的新列表进行比较，以创建一个嵌套列表，该列表根据它们的订单号分隔文件。使用相同的示例目录，列表将如下所示：[[1234567890-UP.csv,1234567890-DOWN.csv],[2000005001-UP.csv,2000005001-DOWN.csv]]

最后，我想遍历此列表并.csv根据列表索引将文件合并在一起。

可能有一种更简单的方法可以做到这一点，但我忽略了它可以为我省去很多麻烦。

我当前使用嵌套 for 循环填充嵌套列表的代码如下所示

    nestedlist=[]
    for x in range(len(filenamelist)):
        for y in range(len(filteredlist)):
            if filteredlist[y] in filenamelist[x]:
                nestedlist[y].append(filenamelist[x]

这将返回一个错误Index out of range。这是因为nestedlist从未定义过的大小。我不确定该怎么做，或者最好的方法是什么？

标签： pythonlistloopsnested

解决方案

这可以通过更简单的方式实现。假设您有目录的文件列表 -

files = [1234567890-UP.csv,1234567890-DOWN.csv,2000005001-UP.csv,2000005001-DOWN.csv]

您可以对此进行迭代，创建订单号到实际文件名的映射。

filemap = {}
for file in files:
  order_number = re.compile('(\d*)-(\w*).csv').match(file).groups()[0]
  print(order_number)
  files = filemap.get(order_number, [])
  files.append(file)
  filemap[order_number] = files

那应该给出这样的东西。

{'1234567890': ['1234567890-UP.csv', '1234567890-DOWN.csv'],
 '2000005001': ['2000005001-UP.csv', '2000005001-DOWN.csv']}

现在您可以查找订单号并在需要时合并

推荐阅读