How to use conditional if/else for multiple dataframe python

Question

I've been been trying the lat and long of dataframe using geocoder.

I managed to get the lat/long of distict city and save it into a file. However, I got an error when try to map the data to other data of city as Im not sure how to do so.

My code:

def func(a):
    if a in data2['CITY']:
        return data2['LATITUDE']
    else:
        return "0"

data["LATITUDE"] = data["CITY"].apply(lambda x: func(x))

Sidenote: lambda x: func(x) is redundant. Just use func. This is called eta-reduction btw. — wjandrea
– wjandrea, Commented Jun 23, 2021 at 16:11
With data2 = pd.to_excel(r'/'), I get AttributeError: module 'pandas' has no attribute 'to_excel'. Maybe you meant pd.read_excel()? But then you'd get IsADirectoryError: [Errno 21] Is a directory: '/'. Please provide a minimal reproducible example. — wjandrea
– wjandrea, Commented Jun 23, 2021 at 16:17
Can you post the output of print(deduplokasi.head().to_dict() and print(data2.head().to_dict()? — not_speshal
– not_speshal, Commented Jun 23, 2021 at 16:24

tdy · Accepted Answer · 2021-06-23 19:55:05Z

1

If I understand correctly, you can generate data['LATITUDE'] by reindexing.

Set data2's index to CITY and reindex() against data's CITY:

data['LATITUDE'] = data2.set_index('CITY').reindex(data['CITY'])['LATITUDE'].values

answered Jun 23, 2021 at 19:55

tdy

42k42 gold badges124 silver badges125 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Dharman · Accepted Answer · 2021-06-23 19:21:25Z

1

In pandas, if you check if a certain element is in a pd.Series, you need to first get it to an array form:

'SCOTLAND' in data2['CITY'] -> False
'SCOTLAND' in data2['CITY'].values -> True

After making this fix, you need to make it so that it returns a single element, not a vector. So, finally:

def func(a):
    if a in data2['CITY'].values:
        return data2.loc[data2['CITY'] == a, 'LATITUDE'].item()
    else:
        return "0"

edited Jun 23, 2021 at 19:21

Dharman♦

33.9k27 gold badges106 silver badges157 bronze badges

answered Jun 23, 2021 at 19:16

pzmn

1545 bronze badges

Collectives™ on Stack Overflow

How to use conditional if/else for multiple dataframe python

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related