Using Pandas to load and read excel data

Question

I am trying to write code to load an excel file that will return:

the entire data frame
total number of rows
total number of columns

I wrote the following code but it gave an error: TypeError: cannot unpack non-iterable int object Can you please help me with this, thank you!

def load_data():

    import pandas as pd  
    df = pd.read_csv("myfile.csv") 
    num_rows = df.shape[0]  
    num_cols = df.shape[1]

    return num_rows
    return num_cols
df, num_rows, num_cols = load_data()

There is no point in having two return statements like that. The second one will never run. What IDE are you using, that wouldn't point that out? Don't put your import inside the function. Also, is there any point in making this a function in the first place? — AMC
– AMC, Commented Nov 24, 2019 at 20:20

Patrick Flynn · Accepted Answer · 2019-11-24 20:20:01Z

1

Try this!

import pandas as pd   

def load_data():

    df = pd.read_csv("myfile.csv") 
    num_rows = df.shape[0]  
    num_cols = df.shape[1]

    return df, num_rows, num_cols

df, num_rows, num_cols = load_data()

you have to return your variables all at once. Best of luck!

answered Nov 24, 2019 at 20:20

Patrick Flynn

814 bronze badges

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

Comments

Andy L. · Accepted Answer · 2019-11-24 20:18:41Z

1

You return it wrong. You need to return a tuple as follows

def load_data():

    import pandas as pd  
    df = pd.read_csv("myfile.csv") 
    num_rows = df.shape[0]  
    num_cols = df.shape[1]

    return df, num_rows, num_cols

df, num_rows, num_cols = load_data()

answered Nov 24, 2019 at 20:18

Andy L.

25.3k4 gold badges20 silver badges30 bronze badges

Comments

AMC · Accepted Answer · 2019-11-24 20:23:56Z

0

A few things:

The second return statement will never be run.
Don't put import statements inside a function.
I don't see much of a reason to even use a function for this.
That isn't an excel file, it's just CSV.

import pandas as pd

df = pd.read_csv('myfile.csv')
num_rows, num_cols = df.shape

Let me know if you have any questions :)

answered Nov 24, 2019 at 20:23

AMC

2,6977 gold badges15 silver badges35 bronze badges

Collectives™ on Stack Overflow

Using Pandas to load and read excel data

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related