Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- df = pd.read_csv(s_root + s_file,sep=";",skiprows=1,header=None)
- df = df.iloc[:-1,:]
- df[1] = df[1].str.strip()
- #######################
- # tranforma tudo em string
- df[6] = [str(x) for x in df[6]]
- #######################
- for x in df[1].unique():
- try:
- dfaux = df.loc[df[1]==x,:]
- #######################
- # filtra string. Note que mudei o df para dfaux tb
- dfaux = dfaux.loc[dfaux[6]=='1',:]
- # nao processa se for matriz em branco
- if dfaux.shape[0] == 0:
- continue
- #######################
- dfaux[2] = dfaux[2].astype(int)
- dfaux[4] = dfaux[4].astype(int)
- dfaux[16] = dfaux[16].astype(int)
- dfaux[17] = dfaux[17].astype(int)
- dfaux[18] = 0
- dfaux.loc[dfaux[10]==1,18]=1
- dfaux.loc[dfaux[14]==1,18]=2
- dfaux.loc[dfaux[15]==1,18]=3
- print 'Step 1='+ dfaux.iloc[0,1]
- dfaux = dfaux.loc[:,[1,0,5,2,3,4,16,17,18]]
- dfaux = dfaux.sort_values(by=2).reset_index(drop=True)
- print 'Step 2='+dfaux.iloc[0,1]
- except Exception as e:
- print e
- print x
- print dfaux
- raise
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement