python读取文件名及后缀 python文件的后缀

Python读取文件名及后缀

在Python中，要读取文件名及后缀，可以使用os模块中的os.path.splitext()方法。os.path.splitext()方法可以分离文件名和扩展名。例如：如果输入的文件名是test.txt，os.path.splitext()方法返回的值是('test', '.txt')。

示例代码：

      import os

      file_path='/path/to/test.txt'
      file_name, file_ext=os.path.splitext(file_path)

      print(file_name) # 输出：test
      print(file_ext) # 输出：.txt
    

在上述示例代码中，os.path.splitext()方法将文件路径分离成文件名和扩展名，分别赋值给变量file_name和file_ext。之后，我们可以打印这两个变量的值，以获得文件名和扩展名。

Python获取文件的后缀

如果我们只需要获取文件的后缀，可以使用字符串的split()方法。以test.txt为例，我们可以使用split()方法将其分割成"test"和"txt"两个部分，然后获取第二个部分，即文件的扩展名。注意：split(".")中的"."是分隔符，不是扩展名的一部分。

示例代码：

      file_name='test.txt'
      file_ext=file_name.split(".")[1]

      print(file_ext) # 输出：txt
    

在上述示例代码中，我们先定义一个文件名file_name，然后使用split()方法将其分割成文件名和扩展名。由于我们只需要获取扩展名，因此我们使用"[1]"获取分割后的第二个部分，即扩展名。

Python获取某个文件夹中所有文件的后缀

如果我们需要获取某个文件夹中所有文件的后缀，可以使用os模块中的os.listdir()方法和字符串的split()方法。os.listdir()方法返回目录中所有文件的文件名，我们可以遍历这些文件名，然后使用split()方法获取每个文件的扩展名。

示例代码：

      import os

      dir_path='/path/to/dir'
      file_ext_list=[]

      for file_name in os.listdir(dir_path):
        file_ext=file_name.split(".")[-1]
        file_ext_list.append(file_ext)

      print(file_ext_list) # 输出：['txt', 'jpg', 'pdf', ...]
    

在上述示例代码中，我们使用os.listdir()方法获取目录dir_path中所有文件的文件名，然后遍历这些文件名，使用split()方法获取每个文件的扩展名，将其添加到file_ext_list列表中。最后，我们可以打印出整个列表。

总结

Python中获取文件名及后缀的方法很简单。如果我们只需要获取文件名和扩展名，可以使用os模块中的os.path.splitext()方法。如果我们只需要获取文件的扩展名，可以使用字符串的split()方法。如果我们需要获取某个文件夹中所有文件的后缀，可以使用os模块中的os.listdir()方法和字符串的split()方法。