--- a
+++ b/benchmark/ccs_utils.py
@@ -0,0 +1,115 @@
+import pandas as pd 
+
+'''
+14 columns
+ICD-10-CM Code	ICD-10-CM Code Definition	Beta Version CCS Category	Beta Version CCS Category Description	CCSR1	CCSR1 Label	CCSR2	CCSR2 Label	CCSR3	CCSR3 Label	CCSR4	CCSR4 Label	CCSR5	CCSR5 Label
+
+'''
+def rawfile2dict():
+	'''
+		ccs beta code
+	'''
+	file = "icdcode/DXCCSR-vs-Beta-CCS-Comparison.xlsx"
+	icd2ccs_file = "icdcode/icd2ccs.txt"
+	ccscode2description_file = 'icdcode/ccs2description.txt'
+	contents = pd.read_excel(open(file, 'rb'), sheet_name = 'ICD-10-CM Code Detail')
+	N, _ = contents.shape 
+
+	icd10code_lst = contents['ICD-10-CM Code']
+	ccs_beta_lst = contents['Beta Version CCS Category']
+	ccs_description_lst = contents['Beta Version CCS Category Description']
+	icd2ccs = dict()
+	ccscode2description = dict() 
+	for i in range(N):
+		icdcode = icd10code_lst[i]
+		ccscode = ccs_beta_lst[i]
+		ccs_description = ccs_description_lst[i]
+		icd2ccs[icdcode] = ccscode 
+		ccscode2description[ccscode] = ccs_description
+	with open(icd2ccs_file, 'w') as fout:
+		for k,v in icd2ccs.items():
+			fout.write(str(k) + '\t' + str(v) + '\n')
+	with open(ccscode2description_file, 'w') as fout:
+		for k,v in ccscode2description.items():
+			fout.write(str(k) + '\t' + str(v) + '\n')
+	return icd2ccs, ccscode2description 
+
+
+def rawfile2dict_CCSR():
+	file = "icdcode/DXCCSR-vs-Beta-CCS-Comparison.xlsx"
+	icd2ccsr_file = "icdcode/icd2ccsr.txt"
+	contents = pd.read_excel(open(file, 'rb'), sheet_name = 'ICD-10-CM Code Detail')
+	N, _ = contents.shape 
+
+	icd10code_lst = contents['ICD-10-CM Code']
+	ccsr_lst = contents['CCSR1']
+	icd2ccsr = dict()
+	for i in range(N):
+		icdcode = icd10code_lst[i]
+		ccsr = ccsr_lst[i][:3]
+		icd2ccsr[icdcode] = ccsr 
+	with open(icd2ccsr_file, 'w') as fout:
+		for k,v in icd2ccsr.items():
+			fout.write(str(k) + '\t' + str(v) + '\n')
+	return icd2ccsr 
+
+
+def file2_icd2ccsr():
+	icd2ccsr_file = "icdcode/icd2ccsr.txt"
+	with open(icd2ccsr_file, 'r') as fin:
+		lines = fin.readlines()	
+	icd2ccsr = {line.split()[0]:line.split()[1] for line in lines}
+	return icd2ccsr
+
+
+def file2_icd2ccs_and_ccs2description():
+	icd2ccs = dict()
+	icd2ccs_file = "icdcode/icd2ccs.txt"
+	ccscode2description_file = 'icdcode/ccs2description.txt'
+	ccscode2description = dict() 
+	with open(icd2ccs_file, 'r') as fin:
+		lines = fin.readlines()
+	icd2ccs = {line.split()[0]:line.split()[1] for line in lines}
+	with open(ccscode2description_file, 'r') as fin:
+		lines = fin.readlines() 
+	ccscode2description = {line.split()[0]:line.split()[1] for line in lines}
+	return icd2ccs, ccscode2description
+
+
+# icd2ccs, ccscode2description = rawfile2dict() 
+
+
+def cancer_filter_icd10code(icd10code):
+	icd2ccs, ccscode2description = file2_icd2ccs_and_ccs2description() 
+	ccs = icd2ccs[icd10code]
+	description = ccscode2description[ccs]
+	return 'cancer' in description.lower() 
+
+
+
+
+
+if __name__ == "__main__":
+	icd2ccsr = rawfile2dict_CCSR() 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+