python - 如何迭代后缀列表以添加到列表中每个变量的末尾？

Question

我想要做的是优化我当前使用数据帧的代码。数据帧遵循类似的命名约定，因此我没有明确命名它们，而是尝试动态创建它们然后处理它们。

days = [1, 2, 3, 4, 5]
#1,2,3,4,5 are dataframes
endings = ['_operation1', '_operation2']
suffixedFrames = [x.add_suffix(y) for x, y in zip(days, endings)]
print(suffixedFrames)

问题是当我运行它时，它只打印出两个数据帧，特别是 1_operation1 和 1_operation2。如何让它在列表日内为每个数据帧添加后缀？

Answer 1

Zip 无法按您的意愿工作，但它会创建一个元组迭代器，其中第 i 个元组包含您正在使用的 2 列表的第 i 个元素，如您所见 in the documentation .在你的情况下，如果你这样做 print(list(zip(days, endings)))你会看到这个:[(1, '_operation1'), (2, '_operation2')] .
为了实现您想要的，您可以执行以下操作:

import itertools
days = [1, 2, 3, 4, 5]
#1,2,3,4,5 are dataframes
endings = ['_operation1', '_operation2']
suffixedFrames = [x.add_suffix(y) for x, y in list(itertools.product(days, endings))]
print(suffixedFrames)