Creating DataFrame from list of lists in Pandas - Visual Walkthrough

Concept Flow - Creating DataFrame from list of lists

Start with list of lists

↓

Pass list to pd.DataFrame()

↓

Optional: Add column names

↓

DataFrame created with rows and columns

↓

Use DataFrame for analysis or display

We start with a list of lists, pass it to pandas DataFrame constructor, optionally add column names, and get a table-like DataFrame.

Execution Sample

Pandas

import pandas as pd

data = [[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]
df = pd.DataFrame(data, columns=['ID', 'Name'])
print(df)

This code creates a DataFrame from a list of lists with two columns: ID and Name, then prints it.

Execution Table

Step	Action	Input Data	DataFrame State	Output
1	Define list of lists	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]	No DataFrame yet	List ready
2	Call pd.DataFrame(data)	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]	DataFrame created with default columns [0, 1]	DataFrame with 3 rows, 2 columns
3	Add columns=['ID', 'Name']	Columns specified	DataFrame columns renamed to ID and Name	DataFrame with columns ID, Name
4	Print DataFrame	DataFrame with data and columns	DataFrame displayed	ID Name 0 1 Alice 1 2 Bob 2 3 Charlie

💡 DataFrame created and printed successfully

Variable Tracker

Variable	Start	After Step 1	After Step 2	After Step 3	Final
data	undefined	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]	[[1, 'Alice'], [2, 'Bob'], [3, 'Charlie']]
df	undefined	undefined	DataFrame with columns [0, 1] and 3 rows	DataFrame with columns ['ID', 'Name'] and 3 rows	DataFrame with columns ['ID', 'Name'] and 3 rows

Key Moments - 2 Insights

Why do we need to specify column names when creating a DataFrame from a list of lists?

Without column names, pandas assigns default numeric column names (0, 1, ...). Specifying columns=['ID', 'Name'] makes the DataFrame easier to read and work with, as shown in step 3 of the execution_table.

What happens if the inner lists have different lengths?

Pandas will fill missing values with NaN for shorter rows. This is not shown in the current example but is important to know when lists vary in length.

Visual Quiz - 3 Questions

Look at the execution_table, what is the DataFrame's columns after step 2?

A ['ID', 'Name']

B [0, 1]

C ['A', 'B']

D No columns assigned

💡 Hint

Check the 'DataFrame State' column in row for step 2 in execution_table

At which step are the column names set to 'ID' and 'Name'?

A Step 1

B Step 2

C Step 3

D Step 4

💡 Hint

Look at the 'Action' and 'DataFrame State' columns in execution_table

If the list had an extra inner list [4, 'Diana'], how would the DataFrame change after step 3?

A It would have 4 rows with IDs 1 to 4

B It would have 3 rows only

C It would cause an error

D It would have 2 columns only

💡 Hint

Adding an extra inner list adds a new row, as shown by the variable 'data' in variable_tracker

Concept Snapshot

Create DataFrame from list of lists:
- Use pd.DataFrame(data) where data is list of lists
- Each inner list is a row
- Optionally add columns=['col1', 'col2'] for names
- Result is a table with rows and columns
- Easy way to convert raw data to DataFrame

Full Transcript

We start with a list of lists representing rows of data. Passing this list to pandas DataFrame constructor creates a table with default numeric column names. We can specify column names to make the table clearer. The DataFrame can then be used for analysis or display. The execution table shows each step from defining the list, creating the DataFrame, adding columns, and printing the result. Variables 'data' and 'df' change accordingly. Key moments clarify why column names matter and what happens with uneven inner lists. The quiz tests understanding of column naming and data structure changes.

Practice

1. What does creating a DataFrame from a list of lists in pandas mainly help you do?

easy

A. Organize data into rows and columns for easy analysis

B. Run SQL queries directly on the list

C. Convert data into images automatically

D. Encrypt data for security

Solution

1. Step 1: Understand DataFrame purpose

   A DataFrame is a table-like structure with rows and columns to organize data.

2. Step 2: Connect list of lists to DataFrame

   A list of lists can be converted into this table format for easier data handling.

3. Final Answer:

   Organize data into rows and columns for easy analysis -> Option A

4. Quick Check:

   DataFrame = Table format [OK]

Hint: DataFrames = tables; list of lists = rows [OK]

Common Mistakes:

• Thinking DataFrame encrypts data
• Confusing DataFrame with image conversion
• Assuming list of lists runs SQL

2. Which of the following is the correct way to create a DataFrame from a list of lists data = [[1, 2], [3, 4]] with columns named 'A' and 'B'?

easy

A. pd.DataFrame(data, columns=['A', 'B'])

B. pd.DataFrame(data, names=['A', 'B'])

C. pd.DataFrame(data, cols=['A', 'B'])

D. pd.DataFrame(data, header=['A', 'B'])

Solution

1. Step 1: Recall DataFrame column naming syntax

   Use the parameter columns to name columns when creating a DataFrame.

2. Step 2: Check each option

   Only pd.DataFrame(data, columns=['A', 'B']) uses columns=['A', 'B'], which is correct syntax.

3. Final Answer:

   pd.DataFrame(data, columns=['A', 'B']) -> Option A

4. Quick Check:

   Use columns= for column names [OK]

Hint: Use columns= to name DataFrame columns [OK]

Common Mistakes:

• Using names= instead of columns=
• Using cols= or header= which are invalid here
• Forgetting to pass columns parameter

3. What will be the output of the following code?

import pandas as pd
data = [[10, 20], [30, 40]]
df = pd.DataFrame(data, columns=['X', 'Y'])
print(df)

medium

A. X Y
0 20 10
1 40 30

B. X Y
0 10 20
1 30 40

C. 0 1
X 10 20
Y 30 40

D. Error: columns parameter missing

Solution

1. Step 1: Create DataFrame with given data and columns

   The list of lists becomes rows; columns=['X', 'Y'] names columns accordingly.

2. Step 2: Print DataFrame

   Output shows index 0 and 1 with values 10,20 and 30,40 under columns X and Y.

3. Final Answer:

   X Y 0 10 20 1 30 40 -> Option B

4. Quick Check:

   DataFrame prints rows with columns X, Y [OK]

Hint: Rows = inner lists; columns = names list [OK]

Common Mistakes:

• Swapping row and column values
• Confusing index with columns
• Expecting error without columns parameter

4. Identify the error in this code snippet:

import pandas as pd
data = [[1, 2], [3, 4]]
df = pd.DataFrame(data, columns=['A'])
print(df)

medium

A. DataFrame cannot be created from list of lists

B. Data should be a dictionary, not list of lists

C. Missing import statement for pandas

D. The columns list length does not match data inner list length

Solution

1. Step 1: Check data shape and columns length

   Each inner list has 2 elements, but columns list has only 1 name.

2. Step 2: Understand pandas error

   pandas requires columns length to match number of elements per row.

3. Final Answer:

   The columns list length does not match data inner list length -> Option D

4. Quick Check:

   Columns count = data columns count [OK]

Hint: Columns count must match inner list length [OK]

Common Mistakes:

• Ignoring mismatch between columns and data
• Thinking list of lists is invalid input
• Forgetting to import pandas

5. You have sales data as a list of lists: sales = [[100, 200], [150, 250], [300, 400]] representing sales in two regions. How do you create a DataFrame with columns 'Region1' and 'Region2' and add a new column 'Total' that sums both regions for each row?

hard

A. df = pd.DataFrame(sales, columns=['Region1', 'Region2'])
df['Total'] = sales[0] + sales[1]
print(df)

B. df = pd.DataFrame(sales)
df['Total'] = df[0] + df[1]
print(df)

C. df = pd.DataFrame(sales, columns=['Region1', 'Region2'])
df['Total'] = df['Region1'] + df['Region2']
print(df)

D. df = pd.DataFrame(sales, columns=['Region1', 'Region2'])
df['Total'] = df.sum()
print(df)

Solution

1. Step 1: Create DataFrame with correct columns

   Use columns=['Region1', 'Region2'] to name columns properly.

2. Step 2: Add 'Total' column by summing the two region columns

   Use df['Total'] = df['Region1'] + df['Region2'] to sum row-wise.

3. Final Answer:

   df = pd.DataFrame(sales, columns=['Region1', 'Region2']) df['Total'] = df['Region1'] + df['Region2'] print(df) -> Option C

4. Quick Check:

   Sum columns row-wise for total [OK]

Hint: Sum columns with df['Total'] = df['A'] + df['B'] [OK]

Common Mistakes:

• Adding lists instead of DataFrame columns
• Using df.sum() without axis=1 for row sums
• Not naming columns when creating DataFrame