Python提取文件名并保存
在日常的编程过程中，我们经常需要处理文件和文件夹。有时候，我们需要提取文件名并进行保存或其他操作。在本文中，我们将介绍如何使用Python来提取文件名并保存。
1. 获取文件名
首先，我们需要获得文件的完整路径。可以通过用户输入、文件选择对话框或者直接指定路径来获取文件的完整路径。
import os
file_path = input("请输入文件路径：") # 用户输入文件路径
file_name = os.path.basename(file_path)
print("文件名：", file_name)
上述代码中，我们使用了Python的os模块来处理文件路径。os.path.basename()函数可以获取文件路径中的文件名。
2. 提取文件名和扩展名
如果我们只想提取文件名，而不包括扩展名，可以使用os.path.splitext()函数。
import os
file_path = input("请输入文件路径：") # 用户输入文件路径
file_name, file_ext = os.path.splitext(os.path.basename(file_path))
print("文件名：", file_name)
print("扩展名：", file_ext)
这样，我们就可以分别获取到文件的文件名和扩展名。
3. 保存文件名
接下来，我们可以使用Python来保存文件名。这个过程可以通过创建一个新文件并将文件名写入其中来实现。下面是一个保存文件名的示例代码。
import os
file_path = input("请输入文件路径：") # 用户输入文件路径
file_name = os.path.basename(file_path)
# 创建新文件并写入文件名
with open("file_names.txt", "w") as file:
file.write(file_name)
上述代码中，我们使用了Python的open()函数来创建一个新文件，然后使用write()方法将文件名写入文件中。如果文件已经存在，将会被覆盖。
4. 完整示例
下面是一个完整的示例代码，演示了如何获取文件名并保存。
import os
def extract_and_save_file_name():
file_path = input("请输入文件路径：") # 用户输入文件路径
file_name = os.path.basename(file_path)
# 创建新文件并写入文件名
with open("file_names.txt", "w") as file:
file.write(file_name)
print("文件名已保存到file_names.txt")
if __name__ == "__main__":
extract_and_save_file_name()
通过调用extract_and_save_file_name()函数，我们可以实现文件名的提取和保存。在运行程序后，用户输入文件路径，然后程序将提取文件名并保存到名为file_names.txt的文件中。
总结
通过本文，我们学习了如何使用Python来提取文件名并保存。我们使用了os.path.basename()函数来获取文件路径中的文件名，使用os.path.splitext()函数来分离文件名和扩展名，使用open()函数和write()方法来保存文件名。这些方法可以帮助我们在处理文件时更加方便地进行操作。
希望本文对你有所帮助！
