python - 为什么我使用 rstrip() 得到这些结果？

Question

这个问题已经有答案了:

Remove substring only at the end of string [duplicate] (11 个回答)

已关闭 5 年前。

我想保留不带 .csv 扩展名的文件名，但使用 rstrip('.csv') 会删除以 s 结尾的字符串中的最后一个字母:

data_files = [
    "ap_2010.csv",
    "class_size.csv",
    "demographics.csv",
    "graduation.csv",
    "hs_directory.csv",
    "sat_results.csv"
]

data_names = [name.rstrip('.csv') for name in data_files]

我得到这个结果:

["ap_2010", "class_size", "demographic","graduation","hs_directory", "sat_result"]

字符串人口统计和 sat_results 的末尾 s 已被删除，为什么会发生这种情况？

Answer 1

这是因为 rstrip() 会从字符串末尾单独去除所有字符。

>>> 'abcdxyx'.rstrip('yx')
'abcd'

这将搜索 y 和 x 以从字符串的右侧剥离。如果您想删除 .csv，您可以使用 split 代替。

>>> "ap_2010.csv".split('.')[0]
"ap_2010"

对于文件名，使用函数os.path.splitext也是一个很好的做法:

>>> import os
>>> os.path.splitext('ap_2010.csv')[0]
"ap_2010"