Create a list from rows in Pandas DataFrame | Set 2 (original) (raw)
Last Updated : 29 Jan, 2019
In an earlier post, we had discussed some approaches to extract the rows of the dataframe as a Python's list. In this post, we will see some more methods to achieve that goal.Note : For link to the CSV file used in the code, click here.Solution #1: In order to access the data of each row of the Pandas dataframe, we can use DataFrame.iloc
attribute and then we can append the data of each row to the end of the list.
Python3 1== `
importing pandas as pd
import pandas as pd
Create the dataframe
df = pd.DataFrame({'Date':['10/2/2011', '11/2/2011', '12/2/2011', '13/2/11'], 'Event':['Music', 'Poetry', 'Theatre', 'Comedy'], 'Cost':[10000, 5000, 15000, 2000]})
Print the dataframe
print(df)
`
Output : Now we will use the
DataFrame.iloc
attribute to access the values of each row in the dataframe and then we will construct a list out of it.
Python3 `
Create an empty list
Row_list =[]
Iterate over each row
for i in range((df.shape[0])):
# Using iloc to access the values of
# the current row denoted by "i"
Row_list.append(list(df.iloc[i, :]))
Print the 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. Just like any other Python's list we can perform any list operation on the extracted list.
Python3 `
Find the length of the newly
created list
print(len(Row_list))
Print the first 3 elements
print(Row_list[:3])
`
Output :
Solution #2: In order to access the data of each row of the Pandas dataframe we can use
DataFrame.iat
attribute and then we can append the data of each row to the end of the list.
Python3 1== `
importing pandas as pd
import pandas as pd
Create the dataframe
df = pd.DataFrame({'Date':['10/2/2011', '11/2/2011', '12/2/2011', '13/2/11'], 'Event':['Music', 'Poetry', 'Theatre', 'Comedy'], 'Cost':[10000, 5000, 15000, 2000]})
Create an empty list
Row_list =[]
Iterate over each row
for i in range((df.shape[0])): # Create a list to store the data # of the current row cur_row =[]
# iterate over all the columns
for j in range(df.shape[1]):
# append the data of each
# column to the list
cur_row.append(df.iat[i, j])
# append the current row to the list
Row_list.append(cur_row)
Print the list
print(Row_list)
`
Output :
Python3 `
Find the length of the newly
created list
print(len(Row_list))
Print the first 3 elements
print(Row_list[:3])
`
Output :