python - Python：如何拆分目录中的csv文件的名称

Question

我的目录中有几个文件名为“salary.csv_result.csv”、“cost.csv_result.csv”......等等。如何拆分 CSV 文件的名称？请建议如何更改目录中的所有文件名。我预期的文件名称是

 'salary.csv', 'cost.csv'...etc..

for files in os.listdir(files_dir):
    file_name = files[:files.find('.')]

标签： pythonfile

Answer 1

这段代码会做到这一点。

for files in os.listdir(files_dir): 
  # Better to call this filename than filename
  splits = files.split('.')
  # The assertions are optional
  assert splits[1] == 'csv_result'
  assert splits[2] == 'csv'
  filename = splits[0] +'.csv'
  # ... filename will have the requested form