Python脚本——.csv文件转.txt脚本三则

第一则

"""
批量给指定文件夹内的所有.csv文件增加表头后转成txt
"""

import os

def add_header_and_convert_to_txt(csv_folder, header, output_folder):
    # 获取指定文件夹中的所有.csv文件
    csv_files = [f for f in os.listdir(csv_folder) if f.endswith('.csv')]

    # 遍历每个.csv文件
    for csv_file in csv_files:
        # 构造完整的文件路径
        csv_file_path = os.path.join(csv_folder, csv_file)

        # 构造.txt文件路径
        txt_file_name = os.path.splitext(csv_file)[0] + '.txt'
        txt_file_name = os.path.basename(txt_file_name)
        txt_file_path = os.path.join(output_folder, txt_file_name)

        # 在.csv文件的第一行之前插入表头，并将其保存为.txt文件
        with open(csv_file_path, 'r', encoding='utf-8') as csv_file:
            lines = csv_file.readlines()
            lines.insert(0, header + '\n')

            with open(txt_file_path, 'w', encoding='utf-8') as txt_file:
                txt_file.writelines(lines)

    print('添加表头并转换为.txt文件完成。')

# 使用方法示例：
csv_folder_path = 'your_folder_path'  # 替换为包含多个.csv文件的文件夹路径
header = 'column1,column2,column3,column4,column5,column6,column7,column8,column9,column10,column11,column12,column13,column14,column15,column16,column17,column18,column19,column20'  # 替换为您的表头内容
output_folder_path = 'your_output_folder_path'  # 替换为转换后的文件输出路径

add_header_and_convert_to_txt(csv_folder_path, header, output_folder_path)

第二则

"""
给文件夹内特定的.csv文件添加表头后转成txt
"""
import os

def add_header_and_convert_to_txt(csv_folder, target_files, header):
    # 遍历目标文件夹中的文件
    for file_name in os.listdir(csv_folder):
        file_path = os.path.join(csv_folder, file_name)
        # 检查文件是否为目标 .csv 文件
        if file_name.endswith('.csv') and file_name in target_files:
            # 构造转换后的文件路径
            txt_file_path = os.path.splitext(file_path)[0] + '.txt'
            # 在 .csv 文件的第一行之前插入表头，并将其保存为 .txt 文件
            with open(file_path, 'r', encoding='utf-8') as csv_file:
                lines = csv_file.readlines()
                lines.insert(0, header + '\n')
                with open(txt_file_path, 'w', encoding='utf-8') as txt_file:
                    txt_file.writelines(lines)

    print('添加表头并转换为 .txt 文件完成。')

# 使用方法示例：
csv_folder_path = 'your_folder_path'  # 替换为包含多个 .csv 文件的文件夹路径
target_csv_files = ['file1.csv', 'file2.csv']  # 替换为目标 .csv 文件名列表
header = 'column1,column2,column3,column4,column5'  # 替换为您的表头内容

add_header_and_convert_to_txt(csv_folder_path, target_csv_files, header)

第三则

"""
根据特定日期，在文件夹内获取特定的.csv文件添加表头后转成txt
"""
import os
import re
import datetime

def add_header_and_convert_to_txt(csv_folder, target_date, header):
    # 构造正则表达式模式
    pattern = re.compile(r'\d{8}\.csv$')

    # 遍历目标文件夹中的文件
    for file_name in os.listdir(csv_folder):
        if os.path.isfile(os.path.join(csv_folder, file_name)):
            # 检查文件名是否匹配正则表达式模式
            if pattern.match(file_name):
                # 提取文件名中的日期部分，并将其转换为 datetime 类型
                date_str = file_name.split('.')[0][-8:]
                file_date = datetime.datetime.strptime(date_str, '%Y%m%d').date()
                # 比较文件日期和目标日期
                if file_date == target_date:
                    # 如果日期匹配，在 .csv 文件的第一行之前插入表头，并将其保存为 .txt 文件
                    csv_file_path = os.path.join(csv_folder, file_name)
                    txt_file_path = os.path.splitext(csv_file_path)[0] + '.txt'
                    with open(csv_file_path, 'r', encoding='utf-8') as csv_file:
                        lines = csv_file.readlines()
                        lines.insert(0, header + '\n')
                        with open(txt_file_path, 'w', encoding='utf-8') as txt_file:
                            txt_file.writelines(lines)
                    print(f'已处理文件：{file_name}')

    print('添加表头并转换为 .txt 文件完成。')

# 使用方法示例：
csv_folder_path = 'your_folder_path'  # 替换为包含多个 .csv 文件的文件夹路径
target_date_str = '20230101'  # 替换为您期望的日期字符串
header = 'column1,column2,column3,column4,column5'  # 替换为您的表头内容

target_date = datetime.datetime.strptime(target_date_str, '%Y%m%d').date()  # 将日期字符串转换为 datetime.date 类型
add_header_and_convert_to_txt(csv_folder_path, target_date, header)

for循环的两种不常见用法

for num in reversed(range(0, 31)):
    print(num)

for num in range(30, 0, -1):
    print(num)