csv新增一列

# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年06月22日19:09
"""
import pandas as pd  # pandas是一个强大的分析结构化数据的工具集
import os


raw_path = 'D:/shujuji/k'
files = os.listdir(raw_path)
for file in files:
    # path=os.path.join(raw_path, file)
    path = raw_path + '/' + file
    file_ = os.listdir(path)
    for i in file_:
        final_path = path + '/' + i
        print(final_path)
        data = pd.read_csv(final_path)
        # 添加新列‘名字长度’（length）
        Label = []  # 准备一个列表，把新列的数据存入其中
        datasize = len(data)
        if 'D01' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(1)
        if 'D02' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(2)
        if 'D03' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(3)
        if 'D04' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(4)
        if 'D05' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(5)

        if 'D06' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(6)
        if 'D19' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(7)
        if 'F' in final_path:
            for index in range(datasize):  # 遍历数据表，计算每一位名字的长度
                Label.append(0)

        data['label'] = Label  # 注明列名，就可以直接添加新列
        data.to_csv(final_path, index=False)  # 把数据写入数据集，index=False表示不加索引