深度阅读

How to get file extension in Python?

作者
作者
2023年08月22日
更新时间
20.56 分钟
阅读时间
0
阅读量

It looks like you may have accidentally included search results for a different question in your response. Nonetheless, here’s how you can get file extension from a file path in Python:

To get the file extension from a file path in Python , you can use the os.path.splitext() function. This function splits the file path into a tuple having two values - the file path and the file extension. You can then access the file extension by selecting the second element of the tuple. For example:

import os

file_path = "/path/to/file.txt"
file_extension = os.path.splitext(file_path)[1]

print(file_extension) # Output: .txt

In this example, we define a file path file_path and use os.path.splitext() to split the path into a tuple having two values. We then select the second element of the tuple to get the file extension and store the result in the variable file_extension. Finally, we print out the file extension to the console.

Note that the file extension includes the leading period (.). If you want to get the file extension without the period, you can use string slicing to remove it.

To traverse a directory in Python and filter files based on their extension

To traverse a directory in Python and filter files based on their extension, you can use the os module and a list comprehension. Here’s an example:

import os

dir_path = '/path/to/directory'
extension = '.txt'

files = [f for f in os.listdir(dir_path) if f.endswith(extension)]

print(files)

In this example, we define a directory path dir_path and an extension extension (in this case, “.txt”). We then use a list comprehension to iterate through all files in the directory and select only those that end with the specified extension. The resulting list of selected files is stored in the files variable and printed to the console.

You can modify the code to choose a different directory and file extension by changing the values of dir_path and extension.

相关标签

博客作者

热爱技术,乐于分享,持续学习。专注于Web开发、系统架构设计和人工智能领域。