如果其中有的文件时CSV文件格式,有的是二进制格式,有的是TXT格式,有的是.py格式,请你完善以下代码,以便能够打印文件的第一行 import os # 一级文件夹路径 root_dir = '/path/to/root/directory' # 遍历一级文件夹下的所有二级文件夹 for subdir in os.listdir(root_dir): # 拼接二级文件夹路径 sub_dir_path = os.path.join(root_dir, subdir) # 判断是否为文件夹 if os.path.isdir(sub_dir_path): # 遍历二级文件夹下的所有文件 for filename in os.listdir(sub_dir_path): # 拼接文件路径 file_path = os.path.join(sub_dir_path, filename) # 判断是否为文件 if os.path.isfile(file_path): # 打印文件路径和表头 with open(file_path, 'r') as f: print('File Path:', file_path) print('Header:', f.readline().strip())
时间: 2024-02-26 19:55:01 浏览: 196
import os
root_dir = '/path/to/root/directory'
for subdir in os.listdir(root_dir):
sub_dir_path = os.path.join(root_dir, subdir)
if os.path.isdir(sub_dir_path):
for filename in os.listdir(sub_dir_path):
file_path = os.path.join(sub_dir_path, filename)
if os.path.isfile(file_path):
with open(file_path, 'rb') as f:
print('File Path:', file_path)
if filename.endswith('.csv'): # 如果是csv文件
print('Header:', f.readline().decode('utf-8').strip()) # 读取并解码第一行
elif filename.endswith('.txt'): # 如果是txt文件
print('Header:', f.readline().strip()) # 读取第一行
elif filename.endswith('.py'): # 如果是py文件
print('Header:', f.readline().strip()) # 读取第一行
elif filename.endswith('.bin'): # 如果是二进制文件
header = f.read(10) # 读取前10个字节作为表头
print('Header:', header)
阅读全文