Python: Looping through Pandas DataFrame to match string in a list

Question

My question is regarding a Pandas DataFrame and a list of e-mail addresses. The simplified dataframe (called 'df') looks like this:

   Name    Address         Email
0  Bush    Apple Street
1  Volt    Orange Street
2  Smith   Kiwi Street

The simplified list of e-mail addresses looks like this:

list_of_emails = ['[email protected]', '[email protected]', '[email protected]']

Is it possible to loop through the dataframe, to check if a last name is (part of) a e-mail address AND then add that email address to the dataframe? The following code does not work unfortunately, because of line 2 I think:

for index, row in df.iterrows():
    if row['Name'] in x for x in list_of_emails:
        df['Email'][index] = x

Your help is very much appreciated!

DeepSpace · Accepted Answer · 2017-07-31 06:57:06Z

4

Generally you should consider using iterrows as last resort only.

Consider this:

import pandas as pd

df = pd.DataFrame({'Name': ['Smith', 'Volt', 'Bush']})

list_of_emails = ['[email protected]', '[email protected]', '[email protected]']

def foo(name):
    for email in list_of_emails:
        if name.lower() in email:
            return email

df['Email'] = df['Name'].apply(foo)

print(df)

#     Name                 Email
# 0  Smith   [email protected]
# 1   Volt  [email protected]
# 2   Bush        [email protected]

answered Jul 31, 2017 at 6:57

DeepSpace

82.1k12 gold badges119 silver badges166 bronze badges

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

1 Comment

PythonSherpa Over a year ago

Thanks @DeepSpace! Apply is the function I was looking for!

Zero · Accepted Answer · 2017-07-31 07:23:37Z

3

Here's one way using apply and lambda function

For, first match

In [450]: df.Name.apply(
           lambda x: next((e for e in list_of_emails if x.lower() in e), None))
Out[450]:
0     [email protected]
1    [email protected]
2          [email protected]
Name: Name, dtype: object

For all matches, in a list

In [451]: df.Name.apply(lambda x: [e for e in list_of_emails if x.lower() in e])
Out[451]:
0     [[email protected]]
1    [[email protected]]
2          [[email protected]]
Name: Name, dtype: object

answered Jul 31, 2017 at 7:23

Zero

77.4k22 gold badges153 silver badges153 bronze badges

Collectives™ on Stack Overflow

Python: Looping through Pandas DataFrame to match string in a list

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related