1. Python
  2. Create list from strings in Pandas dataframe

Create list from strings in Pandas dataframe



Solution # 1: To iterate over rows in DataFrame.iterrows () Pandas we can use DataFrame.iterrows () and then add the data of each row to the end of the list.

# import pandas as pd

import pandas as pd

  
# Create data frame

df = pd.DataFrame ({ ` Date` : [ ` 10/2 / 2011` , `11/2 / 2011` , `12/2 / 2011` , ` 13/2 / 11` ],

`Event` : [ `Music` , ` Poetry` , `Theater` , ` Comedy ` ],

  ` Cost` : [ 10000 , 5000 , 15000 , 2000 ]})

 
# Print data frame

print (df)

Output:

 

We will now use DataFrame.iterrows () to iterate over each row of the given Dataframe and building a list from the data of each row.

# Create empty list

Row_list = []

 
# Iterate over each line

for index, rows in df.iterrows ():

# Create a list for the current line

my_list = [rows.Date, rows.Event, rows.Cost]

 

# add list to final list

  Row_list.append (my_list)

 
# Print list

print (Row_list)

Output:

As we can see in the output, we have successfully extracted each row of a given dataframe into a list. Like any other Python list, we can perform any operation on the list of the retrieved list.

# Find the length again
# list created

print ( len (Row_list))

  
# Print first 3 items

print (Row_list [: 3 ])

Output:

Solution # 2. To iterate over the lines in DataFrame.itertuples () Pandas, we can use DataFrame.itertuples () and then add each row`s data to the end of the list.

# import pandas as pd

import pandas as pd

 
# Create data frame

df = pd.DataFrame ({ `Date ` : [ ` 10/2 / 2011` , `11/2 / 2011` , ` 12/2 / 2011` , `13/2 / 11` ],

`Event` : [ ` Music` , `Poetry` , ` Theater` , `Comedy` ],

  ` Cost` : [ 10000 , 5000 , 15000 , 2000 ]})

 
# Print the data frame

print (df)

Output:

We will now use DataFrame.itertuples () to iterate over each row of the given Dataframe and create a list from each row`s data.

# Create empty list

Row_list = []

 
# Iterate over each row

for rows in df.itertuples ():

# Create a list for the current line

my_list = [rows.Date, rows.Event, rows.Cost]

 

# add the list to final list

Row_list.append (my_list)

 
# Print list

print (Row_list)

Output:

As we can see in the output, we have successfully extracted each row of the given dataframe into a list. Like any other Python list, we can perform any operation on the list of the retrieved list.

# Find the length again
# list created

print ( len (Row_list))

  
# Print first 3 items

print (Row_list [: 3 ])

Output: