python 数据预处理之打标签

分享一个用python对excel数据指定字段进行打标签的小代码

原数据：

处理后的数据：

 代码：

import openpyxl

#加载相应excel文件
exl=openpyxl.load_workbook('D://hk//spark课设//cars_pre.xlsx')
sheet=exl[exl.sheetnames[0]]
sheet=exl.active

#kinds为二维列表，存储需要打标签的字段各个标签对应值在excel表中的所有行索引，fun函数为执行单元格修改操作的主函数
def fun(col,kinds):
    for i,e in enumerate(kinds):
        for j in e:
            sheet[col+str(j)]=i

#集合k收集指定字段的各个标签对应值，pre函数收集并返回kinds列表
def pre(col):
    k,kinds=set(),[]
    for row in range(sheet.max_row-1):
        k.add(sheet.cell(row=row+2,column=col).value)

    print(len(k),k)

    for i in k:
        kinds.append([])
        for row in range(sheet.max_row-1):
            if sheet.cell(row=row+2,column=col).value==i:
                kinds[-1].append(row+2)
    print(kinds)
    return kinds

fun('b',pre(2))
fun('g',pre(7))
fun('i',pre(9))
fun('j',pre(10))
fun('k',pre(11))

exl.save('D://hk//spark课设//cars_pre.xlsx')