btstest1 / data_process_func.py
potatoslc's picture
Upload 2 files
3b49b6d
Raw
History Blame Contribute Delete
2.17 kB
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon May 8 20:35:33 2023
@author: potato
"""
import pandas as pd
import os
def df_to_dict_univ_region(df)->dict:
result_dict = {}
sz1,sz2 = df.shape
for i in range(sz1):
if(df.iloc[i,0] in result_dict):
result_dict[df.iloc[i,0]].add(df.iloc[i,1])
else:
temp_set = set()
result_dict[df.iloc[i,0]] = temp_set
result_dict[df.iloc[i,0]].add(df.iloc[i,1])
return result_dict
def df_to_dict_topic_qlist(df)->dict:
result_dict = {}
df_header = list(df.columns)
sz1,sz2 = df.shape
for i in range(sz2-1):
temp_dict = {}
#temp_name = df_header[i+1]
for j in range(sz1):
temp_set = parse_to_set(df.iloc[j,i+1])
temp_dict[df.iloc[j,0]] = temp_set
result_dict[df_header[i+1]]=temp_dict
return result_dict
def df_to_dict_qlist(df)->dict:
result_dict = {}
sz1,_ = df.shape
for i in range(sz1):
result_dict[df.iloc[i,0]] = df.iloc[i,1]
return result_dict
def parse_to_set(ostr:str)->set:
result_list = ostr.split(",")
result_set = set(result_list)
return result_set
def display_dict(dict1:dict)->dict:
result_dict = {}
for k,_ in dict1.items():
result_dict[k] = k
return result_dict
def list_display(list1)->dict:
result_dict = {}
for i in range(len(list1)):
result_dict[list1[i]] = list1[i]
return result_dict
def get_qlist_by_cate(qlist:list,qlist_qcontent_dict):
result_dict = {}
for i in range(len(qlist)):
result_dict[qlist_qcontent_dict[qlist[i]]]= qlist_qcontent_dict[qlist[i]]
return result_dict
def get_v_count(df):
return
def get_univreg_by_school_dict(df):
sz1,_ = df.shape
lea_dict = {}
for i in range(sz1):
if df.iloc[i,1] not in lea_dict:
lea_dict[df.iloc[i,1]]= df.iloc[i,0]
return lea_dict