Как исправить KeyError при назначении регионов состояниям в словаре - PullRequest
2 голосов
/ 17 июня 2019

Я назначаю регионы для каждого отдельного штата. Мой код читает из файла Excel, и там около 30 тыс. Строк. Я установил словарь, присваивающий каждому государству регион, а также сокращения штатов для каждого названия штата. Я пытаюсь создать столбец, который будет заполнять область каждого элемента строки, но получаю сообщение об ошибке KeyError со значением «MA» (поскольку в моем файле Excel нет позиций для этого состояния).

Я попытался написать фразу, используя «кроме», а также «если отсутствует», и ни одна из них не устраняет ошибку и не дает желаемых результатов. Я также попытался удалить MA из словаря, но появляется та же ошибка. Я новичок в Python - я уверен, что это легко исправить, но я не знаю, что это такое.

states = {
    'AK': 'Alaska',
    'AL': 'Alabama',
    'AR': 'Arkansas',
    'AZ': 'Arizona',
    'CA': 'California',
    'CO': 'Colorado',
    'CT': 'Connecticut',
    'DC': 'District of Columbia',
    'DE': 'Delaware',
    'FL': 'Florida',
    'GA': 'Georgia',
    'HI': 'Hawaii',
    'IA': 'Iowa',
    'ID': 'Idaho',
    'IL': 'Illinois',
    'IN': 'Indiana',
    'KS': 'Kansas',
    'KY': 'Kentucky',
    'LA': 'Louisiana',
    'MA': 'Massachusetts',
    'MD': 'Maryland',
    'ME': 'Maine',
    'MI': 'Michigan',
    'MN': 'Minnesota',
    'MO': 'Missouri',
    'MS': 'Mississippi',
    'MT': 'Montana',
    'NC': 'North Carolina',
    'ND': 'North Dakota',
    'NE': 'Nebraska',
    'NH': 'New Hampshire',
    'NJ': 'New Jersey',
    'NM': 'New Mexico',
    'NV': 'Nevada',
    'NY': 'New York',
    'OH': 'Ohio',
    'OK': 'Oklahoma',
    'OR': 'Oregon',
    'PA': 'Pennsylvania',
    'RI': 'Rhode Island',
    'SC': 'South Carolina',
    'SD': 'South Dakota',
    'TN': 'Tennessee',
    'TX': 'Texas',
    'UT': 'Utah',
    'VA': 'Virginia',
    'VT': 'Vermont',
    'WA': 'Washington',
    'WI': 'Wisconsin',
    'WV': 'West Virginia',
    'WY': 'Wyoming'
}
stateplusdc = states.keys()
state_abbrev = {v: k for k, v in states.items()}
state_code = {
    'AK': '10','AL': '4', 'AR': '9', 'AR': '6', 'CA': '9', 'CO': '8',   
'CT': '1', 'DC': '3', 'DE': '3', 'FL': '4', 'GA': '4', 'HI': '9', 'IA': '7', 'ID': '10', 'IL': '5', 'IN': '5', 'KS': '7', 'KY': '4', 'LA': '6', 'MA': '1', 'MD': '3', 'ME': '1', 'MI': '5', 'MN': '5','MO': '7', 'MS': '4', 'MT': '8', 'NC': '4', 'ND': '8', 'NE': '7', 'NH': '1', 'NJ': '2', 'NM': '6','NV': '9', 'NY': '2', 'OH': '5', 'OK': '6','OR': '10', 'PA': '3', 'PR': '2', 'RI': '1', 'SC': '4', 'SD': '8', 'TN': '4', 'TX': '6', 'UT': '8', 'VA': '3', 'VI': '2', 'VT': '1', 'WA': '10', 'WI': '5', 'WV': '3', 'WY': '8', 'PI': '9'
}

state_region = {v: k for k, v in state_code.items()}

excel_file = r'/Users/amandawhiting/Desktop/PA_spending_excel.xlsx'
df = pd.read_excel(excel_file)
df = df.rename(columns={'DAMAGE_CATEGORY_CODE': 'damageCode', 'FEDERAL_SHARE_OBLIGATED':'FedShareObligated',  'PROJECT_AMOUNT': 'ProjectAmount'})  
df = df[df['FedShareObligated']>= 0] 
df = df[df['ProjectAmount'] >= 0df2 =   pd.read_csv("/Users/amandawhiting/Desktop/DisasterDeclarationsSummaries.csv", usecols = ['disasterNumber', 'fyDeclared', 'state'])

df = df[df['damageCode'] != 'A - Debris Removal']
df = df[df['damageCode'] != 'B - Protective Measures']
df = df[df['damageCode'] != 'Z - State Management']
df = df[df['damageCode'] != 'H - Fire Management']
df = df.drop_duplicates() 
df = df.reset_index(drop=True)

df2 = pd.read_csv("/Users/amandawhiting/Desktop/DisasterDeclarationsSummaries.csv", usecols = ['disasterNumber', 'fyDeclared', 'state'])
df2 = df2[df2['fyDeclared'] > 1991]
df2 = df2[df2['fyDeclared'] < 2017]
df2 = df2.reset_index(drop=True)
df2['disasterNumber'] = df2['disasterNumber'].astype(int)

fulldf = pd.merge(df, df2, left_on = 'DISASTER_NUMBER', right_on = 'disasterNumber', how = 'inner',)
fulldf = fulldf.drop_duplicates() 
fulldf = fulldf.reset_index(drop=True)

def get_region():
    return [state_region[i] for i in fulldf['state']]

fulldf["Region"] = get_region()

fulldf.head()

Ожидаемые результаты: Новый столбец в существующей таблице с меткой «Регионы», который заполняет каждую ячейку соответствующей областью для этого состояния в этой строке.

Фактические результаты: Ошибка ключа «MA»:


KeyError                                  Traceback (most recent call    last)
<ipython-input-403-13becd272809> in <module>
 31     return [state_region[i] for i in fulldf['state']]
 32 
    ---> 33 fulldf["Region"] = get_region()
 34 
 35 fulldf.head()

<ipython-input-403-13becd272809> in get_region()
 29 
 30 def get_region():
    ---> 31     return [state_region[i] for i in fulldf['state']]
 32 
 33 fulldf["Region"] = get_region()

<ipython-input-403-13becd272809> in <listcomp>(.0)
 29 
 30 def get_region():
    ---> 31     return [state_region[i] for i in fulldf['state']]
 32 
 33 fulldf["Region"] = get_region()

KeyError: 'MA'

Ответы [ 2 ]

0 голосов
/ 17 июня 2019

IIUC, вы ищете либо:

# replace non-available keys with NA
fulldf["Region"] = fulldf['state'].map(state_region)

или

# keep the non-available keys intact
fulldf["Region"] = fulldf['state'].replace(state_region)
0 голосов
/ 17 июня 2019

Попробуйте:

state_region = {v: k for k, v in state_code.items()}

def get_region():
    result = []
    for i in fulldf['state'] :
        if i in state_region :
            result.append(state_region[i])
        else :
            result.append("NA")
    return result

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...