Возврат нескольких значений в python и добавление их к уникальным столбцам в кадре данных

Question

Справочная информация:

У меня есть функция, которая получает набор атрибутов из базы данных. Вот функция:

def getData(key, full_name, address, city, state, zipcode):
    try:
        url = 'https://personator.melissadata.net/v3/WEB/ContactVerify/doContactVerify'
        payload={
                'TransmissionReference': "test", # used by you to keep track of reference
                'Actions': 'Check',
                'Columns': 'Gender','DateOfBirth','DateOfDeath','EthnicCode','EthnicGroup','Education','PoliticalParty','MaritalStatus','HouseholdSize','ChildrenAgeRange','PresenceOfChildren','PresenceOfSenior','LengthOfResidence','OwnRent','CreditCardUser','Occupation','HouseholdIncome',
                'CustomerID': key,# key
                'Records': [{'FullName': str(full_name), 'AddressLine1': str(address), 'City': str(city), 'State': str(state), 'PostalCode': str(zipcode)}]
                }
        headers = {'Content-Type': 'application/json; charset=utf-8', 'Accept':'application/json', 'Host':'personator.melissadata.net','Expect': '100-continue', 'Connection':'Keep-Alive'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        dom = json.loads(r.text)

        Gender = dom['Records'][0]['Gender']
        DateOfBirth = dom['Records'][0]['DateOfBirth']
        DateOfDeath = dom['Records'][0]['DateOfDeath']
        EthnicCode = dom['Records'][0]['EthnicCode']
        EthnicGroup = dom['Records'][0]['EthnicGroup']
        Education = dom['Records'][0]['Education']
        PoliticalParty = dom['Records'][0]['PoliticalParty']
        MaritalStatus = dom['Records'][0]['MaritalStatus']
        HouseholdSize = dom['Records'][0]['HouseholdSize']
        ChildrenAgeRange = dom['Records'][0]['ChildrenAgeRange']
        PresenceOfChildren = dom['Records'][0]['PresenceOfChildren']
        PresenceOfSenior = dom['Records'][0]['PresenceOfSenior']
        LengthOfResidence = dom['Records'][0]['LengthOfResidence']
        OwnRent = dom['Records'][0]['OwnRent']
        CreditCardUser = dom['Records'][0]['CreditCardUser']
        Occupation = dom['Records'][0]['Occupation']
        HouseholdIncome = dom['Records'][0]['HouseholdIncome']

        return Gender
    except:
        return None

Чтобы создать столбец «Пол», я обертываю функцию в лямбду, как показано ниже:

df['Gender'] = df.apply(lambda row: getData(key, row['Full Name'], row['Address'], row['City'], row['State'], row['Zipcode']))

Цель: Я хочу сделать этот процесс одновременно для всех других атрибутов, которые вы видите ниже Пол, как я могу сделать это в Python.

Answer 1

Вы можете вернуть словарь, а затем развернуть серию объектов словаря:

fields = ['Gender', 'DateOfBirth', etc.]

def getData(key, full_name, address, city, state, zipcode):
    try:
        # your code as before
        dom = json.loads(r.text)
        return {k: dom['Records'][0][k] for k in fields}
    # modify below: good practice to specify exactly which error(s) to catch
    except:
        return {}

Затем расширьте свою серию словарей:

dcts = df.apply(lambda row: getData(key, row['Full Name'], row['Address'], row['City'],
                                    row['State'], row['Zipcode']), axis=1)

df = df.join(pd.DataFrame(dcts.tolist()))

Согласно комментарию @ spaniard, если вы хотите все доступные поля, вы можете просто использовать:

return json.loads(r.text)['Records'][0]