DataFrame按列的unique值划分源数据为unique个子数据DataFrame

最新推荐文章于 2025-04-01 07:49:55 发布

煲饭酱

最新推荐文章于 2025-04-01 07:49:55 发布

阅读量4.8k

点赞数

分类专栏： pandas python基础

本文链接：https://blog.youkuaiyun.com/weixin_40040404/article/details/107031194

版权

dataframe按某一列的distinct值将源数据划分成多个子数据dataframe,并分别将子数据保存为csv

# -*- coding: utf-8 -*-
# @Author   : Administrator
# @DateTime : 2020/6/29 22:35
# @FileName : subdata.py
# @SoftWare : PyCharm

import numpy as np
import os
import pandas as pd

from config.features import branch
from config.params import CSV_DEFAULT_DIR

LABELS = ['X', 'Y', 'Z', 'H']
COLUMNS = ['X', 'Y', 'Z', 'H', 'Addr', 'HTTP_UserAgent',
           'DHCP_Option 55', 'DHCP_Option 60', 'DHCP_HostName',
           'LLDP_System Description', 'LLDP_System Name', 'LLDP_Model Name',
           'MDNS_SRV Service', 'MDNS_Answer Model',
           'CDP_Platform']

FILE_DEFAULT_NAME = 'template.csv'


def generate_csv(