import csvimport os# csv文件所在的文件夹路径folder_path = "/path/to/folder"# 遍历文件夹中的所有csv文件for file_name in os.listdir(folder_path): if file_name.endswith(".csv"): file_path = os.path.join(folder_path, file_name) with open(file_path, "r") as f: reader = csv.reader(f) # 获取第一行的数据 first_row = next(reader) # 获取第5列和第6列的数据作为新文件名 new_file_name = f"{first_row[4]}_{first_row[5]}.csv" # 生成新文件路径 new_file_path = os.path.join(folder_path, new_file_name) # 重命名文件 os.rename(file_path, new_file_path)这段代码如果要加入出现空文件就跳过这个文件怎么改
时间: 2024-02-21 07:01:17 浏览: 25
你可以在读取文件之前,先判断一下文件是否为空,如果为空就跳过这个文件,例如:
```
import csv
import os
# csv文件所在的文件夹路径
folder_path = "/path/to/folder"
# 遍历文件夹中的所有csv文件
for file_name in os.listdir(folder_path):
if file_name.endswith(".csv"):
file_path = os.path.join(folder_path, file_name)
# 判断文件是否为空
if os.stat(file_path).st_size == 0:
print(f"文件 {file_path} 为空,已跳过")
continue
with open(file_path, "r") as f:
reader = csv.reader(f)
# 获取第一行的数据
first_row = next(reader)
# 获取第5列和第6列的数据作为新文件名
new_file_name = f"{first_row[4]}_{first_row[5]}.csv"
# 生成新文件路径
new_file_path = os.path.join(folder_path, new_file_name)
# 重命名文件
os.rename(file_path, new_file_path)
```
在这个代码中,我使用了 `os.stat(file_path).st_size` 来获取文件大小,如果文件大小为 0,就说明这个文件是空的,此时就会跳过这个文件。