fork(1) download
  1. data = """Index CIDADE ESTADO REGIAO CAPITAL POPULACAO_TOTAL_2010
  2. 0 Brejo De Areia MA Centro-Oeste 0 5577.0
  3. 1 Cujubim RO Centro-Oeste 0 15854.0
  4. 2 Balbinos SP Sudeste 0 3702.0
  5. 3 Pacaraima RR Centro-Oeste 0 10433.0
  6. 4 Araquari SC Centro-Oeste 0 24810.0
  7. 5 Pedra Branca Do Amapari AP Centro-Oeste 0 10772.0
  8. 6 Ipiranga Do Norte MT Centro-Oeste 0 5123.0
  9. 7 Parari PB Centro-Oeste 0 1256.0
  10. 8 Colniza MT Centro-Oeste 0 26381.0
  11. 9 Luís Eduardo Magalhães BA Centro-Oeste 0 60105.0"""
  12.  
  13. data = [ i.split("\t") for i in data.split("\n") ] # Transformar em Array de Array
  14.  
  15.  
  16. import pandas as pd
  17.  
  18. df = pd.DataFrame( data[1:], columns=data[0] ) # CARREGAR DADOS
  19.  
  20. del df["Index"] # DELETAR coluna Index
  21. del df['REGIAO'] # DELETAR coluna REGIAO
  22.  
  23.  
  24. regiao = {
  25. "Norte":['AM', 'PA', 'RR', 'AC','RO','AP', 'TO'],
  26. "Sul":['PR', 'SC', 'RS'],
  27. "Centro Oeste":['MT', 'MS', 'GO', 'DF'],
  28. "Nordeste":['BA', 'CE','MA', 'PB', 'PE', 'RN', 'SE', 'AL', 'PI'],
  29. "Sudeste":['SP', 'RJ', 'MG','ES'],
  30. }
  31.  
  32.  
  33. for k in regiao:
  34. df.loc[ df['ESTADO'].isin(regiao[k]), 'REGIAO'] = k
  35.  
  36. print(df)
Success #stdin #stdout 0.48s 60512KB
stdin
Standard input is empty
stdout
                    CIDADE ESTADO CAPITAL POPULACAO_TOTAL_2010        REGIAO
0           Brejo De Areia     MA       0               5577.0      Nordeste
1                  Cujubim     RO       0              15854.0         Norte
2                 Balbinos     SP       0               3702.0       Sudeste
3                Pacaraima     RR       0              10433.0         Norte
4                 Araquari     SC       0              24810.0           Sul
5  Pedra Branca Do Amapari     AP       0              10772.0         Norte
6        Ipiranga Do Norte     MT       0               5123.0  Centro Oeste
7                   Parari     PB       0               1256.0      Nordeste
8                  Colniza     MT       0              26381.0  Centro Oeste
9   Luís Eduardo Magalhães     BA       0              60105.0      Nordeste