Applying a function to pandas dataframe and add a new column?

Question

I got a simple dataframe. It basically looks like this - only much larger.

   import pandas as pd
    csv = [{"name" : "Peters Company", "Apples" : 1}, {"name" : "Quagmires Company", "Apples" : 0}]
    df = pd.DataFrame(csv)

I trying to apply a little function I wrote to the name column. Here is what I do:

from google import search
def get_url(query):
    url = search(query, tld='com', num=1, stop=0, pause=10)
    print(next(url))

I am using google to search for a certain query and print it afterwords. I am trying to create a new column url which contains the result of get_url row by row.

Here is what I did:

for i in df.name:
    get_url(i)

Obviously, this only results in the urlgetting printed one by one. But I trying to expand the dataframe. I tried my luck with itterows and df.locbut so far it didn't work out. Any ideas? Thanks

AndreyF · Accepted Answer · 2017-06-26 11:21:31Z

3

The apply method is exactly what you want. All you need to do is to add a return value to your function:

def get_url(query):
    url = search(query, tld='com', num=1, stop=0, pause=10)
    return next(url) 

df['url'] = df['name'].apply(get_url)

If you want to pass other parameters in addition to the name cell, you can use lambda:

def get_url(query, another_param):
        url = search(query, tld='com', num=1, stop=0, pause=10)
        return next(url) 

df['url'] = df['name'].apply(lambda column_name: get_url(column_name, another_value))

answered Jun 26, 2017 at 11:21

AndreyF

1,8381 gold badge15 silver badges26 bronze badges

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

Comments

jezrael · Accepted Answer · 2017-06-26 11:09:45Z

3

You can use apply:

df['url'] = df['name'].apply(get_url)

Or assign:

df = df.assign(url=df['name'].apply(get_url))

Or list comprehension:

df['url'] = [get_url(x) for x in df['name']]

answered Jun 26, 2017 at 11:09

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

3 Comments

jezrael Over a year ago

Glad can help. Do you use from another answer second solution?

Rachel Over a year ago

The solution I chose added the return() function. It's just a piece I missed...

jezrael Over a year ago

Ok, I missed it also ;) Nice day!

Collectives™ on Stack Overflow

Applying a function to pandas dataframe and add a new column?

2 Answers 2

Comments

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related