📜  pandas 从文件列表中删除扩展名 - Python (1)

📅  最后修改于: 2023-12-03 15:03:29.065000             🧑  作者: Mango

Pandas 从文件列表中删除扩展名 - Python

当我们需要从文件列表中删除文件扩展名时,可以使用 Python 的 os.path.splitext() 方法。但是,如果我们正在使用 pandas,我们可以使用其 str accessor 和 apply() 方法来简化此过程。

以下是使用 pandas 从文件列表中删除扩展名的示例代码:

import pandas as pd
import os

# 创建文件列表
files = ['file1.txt', 'file2.csv', 'file3.xlsx']

# 将列表转换为 pandas 中的 Series
file_series = pd.Series(files)

# 使用 str accessor 和 apply() 方法删除扩展名
file_no_ext = file_series.str.split('.').str[0]

# 打印结果
print(file_no_ext)

输出结果:

0    file1
1    file2
2    file3
dtype: object

在上面的代码中,我们首先将文件列表转换为 pandas 的 Series 对象。然后,我们使用 str accessor 和 apply() 方法来删除每个文件名的扩展名。最后,我们打印结果。

需要注意的是,上述代码仅适用于文件名只包含一个“.”的情况,如果文件名包含多个“.”,则需要进行适当的修改。此外,我们还需要确保文件名中不包含任何其他不正常的符号,因为这可能会影响代码的正确性。

总的来说,使用 pandas 的 str accessor 和 apply() 方法可以使从文件列表中删除扩展名变得更加简单方便。