如何连接字符串拆分中的元素？

Question

非常简单的问题，但我正在努力解决这个问题......如果我有一个名为“first_second_third_fourth.txt”的文件名，并且我想在创建新文件时保留该字符串的前几个元素，我通常会运行如下内容：

import sys
file = sys.argv[1]
new_name = file.split("_")[0] + "_newfile.txt"

output 只是first_newfile.txt

但是，如果我想以某种方式获得一个名为first_second_newfile.txt的新文件，我无法想出一个简单的解决方案来将其写在一行中。

我试过了：

import sys
file = sys.argv[1]
new_name = file.split("_")[0:1] + "_newfile.txt"

但这会引发错误，因为您无法将字符串连接到列表。 我管理的唯一方法是将其拆分，但这似乎很混乱：

import sys
file = sys.argv[1]
new_name = file.split("_")[0] + file.split("_")[1] + "_newfile.txt"

我只是好奇是否有更短的方法来获取拆分列表中的两个元素而无需第二次迭代file.split("_")[1] ？

Answer 1

您可以使用.join()创建所需的文件名：

f = "first_second_third_fourth.txt"
'_'.join(f.split('_')[0:2]) + '.txt

Output：

'first_second.txt'