带有列表的Python列表理解

Question

我正在文件中进行字符串替换，并使用csv writer在输出中写入选项卡。 使用csv编写器意味着我需要一个格式正确的列表来提供给csv编写器以进行输出。 我输入指定选项卡和换行符的输入行，并将其放入csv编写器知道如何写出的列表中。

我下面的代码实际上完成了所需的任务，我想知道的是，是否可以使用for循环上的列表理解功能进一步完善代码，以将选项卡分为自己的列表并替换列表中的现有项。 这是代码：

testInput = "\\t//blah/...\\t//blah_blah/...\\n\\t-//blah/...\\t//blah/...\\n\\t//blah/...\\t//blah_blah/...\\n\\t//blah/...\\t//blah/...\\n"

new_lines = [z for z in testInput.split('\\n') if z]

x=-1
if testInput.find("\\t") >= 0:
    for line in new_lines:
        x+=1
        lines = line.split("\\t")
        new_lines[x] = lines
replacement_value = new_lines

使用以下代码，for for循环将转换new_lines：

对此：

可以使用列表理解重写for循环吗？

Answer 1

我认为您可以在列表组件内使用列表组件来做到这一点...

[[s for s in z.split("\\t")] for z in testInput.split('\\n') if z]

由于我很好奇，看来您可以以这种方式回去...

"\\n".join(["\\t".join([a for a in x]) for x in new_lines])+"\\n"