Converting to and from Python lists in NumPy - Time & Space Complexity

Time Complexity: Converting to and from Python lists

O(n)

Understanding Time Complexity

We want to understand how long it takes to convert between numpy arrays and Python lists.

How does the time grow when the data size gets bigger?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import numpy as np

arr = np.arange(1000)  # Create a numpy array with 1000 elements
py_list = arr.tolist()  # Convert numpy array to Python list
new_arr = np.array(py_list)  # Convert Python list back to numpy array

This code converts a numpy array to a Python list, then back to a numpy array.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Traversing all elements to copy data during conversion.
• How many times: Once for each element when converting to list, and once again when converting back to array.

How Execution Grows With Input

As the number of elements increases, the time to convert grows proportionally.

Input Size (n)	Approx. Operations
10	About 20 operations (10 to list + 10 to array)
100	About 200 operations
1000	About 2000 operations

Pattern observation: The operations roughly double the input size, growing linearly.

Final Time Complexity

Time Complexity: O(n)

This means the time to convert grows directly in proportion to the number of elements.

Common Mistake

[X] Wrong: "Conversion between numpy arrays and lists is instant regardless of size."

[OK] Correct: Each element must be copied, so bigger data takes more time.

Interview Connect

Understanding this helps you explain data handling efficiency and choose the right data structures in real projects.

Self-Check

"What if we convert a nested numpy array (multi-dimensional) to a list? How would the time complexity change?"

Practice

1. What does the function np.array() do when given a Python list?

easy

A. It deletes the Python list.

B. It converts the Python list into a Python dictionary.

C. It converts the Python list into a NumPy array.

D. It converts the Python list into a string.

Solution

1. Step 1: Understand np.array() purpose

   The function np.array() is designed to take a Python list and create a NumPy array from it.

2. Step 2: Compare options with function behavior

   Only It converts the Python list into a NumPy array. correctly states that it converts a list to a NumPy array. Other options describe unrelated actions.

3. Final Answer:

   It converts the Python list into a NumPy array. -> Option C

4. Quick Check:

   np.array(list) = NumPy array [OK]

Hint: np.array() always makes a NumPy array from a list [OK]

Common Mistakes:

• Thinking np.array() converts to dict or string
• Confusing np.array() with list methods
• Assuming np.array() deletes data

2. Which of the following is the correct syntax to convert a NumPy array arr back to a Python list?

easy

A. arr.tolist()

B. arr.to_list()

C. tolist(arr)

D. list(arr.to)

Solution

1. Step 1: Recall the method to convert NumPy array to list

   The correct method is tolist() called on the array object.

2. Step 2: Check each option's syntax

   arr.tolist() uses arr.tolist(), which is correct. arr.to_list() has a wrong method name. Options C and D are invalid syntax.

3. Final Answer:

   arr.tolist() -> Option A

4. Quick Check:

   array.tolist() converts to list [OK]

Hint: Use arr.tolist() to get a Python list from NumPy array [OK]

Common Mistakes:

• Using arr.to_list() instead of arr.tolist()
• Trying to call tolist() as a standalone function
• Incorrect method chaining or syntax errors

3. What is the output of this code?

import numpy as np
lst = [1, 2, 3]
arr = np.array(lst)
print(arr.tolist())

medium

A. [[1], [2], [3]]

B. array([1, 2, 3])

C. TypeError

D. [1, 2, 3]

Solution

1. Step 1: Convert list to NumPy array

   The list [1, 2, 3] is converted to a NumPy array arr with values [1 2 3].

2. Step 2: Convert NumPy array back to list

   Calling arr.tolist() converts the array back to a Python list [1, 2, 3].

3. Final Answer:

   [1, 2, 3] -> Option D

4. Quick Check:

   arr.tolist() = original list [OK]

Hint: arr.tolist() returns the original list values [OK]

Common Mistakes:

• Expecting array output instead of list
• Confusing print output with array representation
• Thinking it returns nested lists

4. Identify the error in this code snippet:

import numpy as np
lst = [1, 2, 3]
arr = np.array(lst)
new_list = arr.to_list()

medium

A. arr.to_list() is not a valid method

B. np.array() cannot convert lists

C. lst should be a NumPy array already

D. Missing import statement

Solution

1. Step 1: Check method names on NumPy arrays

   The method to convert a NumPy array to a list is tolist(), not to_list().

2. Step 2: Identify the error in the code

   Using arr.to_list() causes an AttributeError because the method does not exist.

3. Final Answer:

   arr.to_list() is not a valid method -> Option A

4. Quick Check:

   Use arr.tolist(), not arr.to_list() [OK]

Hint: Use arr.tolist(), not arr.to_list() [OK]

Common Mistakes:

• Typing to_list() instead of tolist()
• Assuming np.array() fails on lists
• Forgetting to import numpy

5. You have a nested Python list nested = [[1, 2], [3, 4]]. Which code correctly converts it to a NumPy array and then back to a nested Python list?

hard

A. arr = np.array(nested)
new_list = list(arr)

B. arr = np.array(nested)
new_list = arr.tolist()

C. arr = list(nested)
new_list = arr.tolist()

D. arr = np.array(nested)
new_list = arr.to_list()

Solution

1. Step 1: Convert nested list to NumPy array

   Using np.array(nested) converts the nested list into a 2D NumPy array.

2. Step 2: Convert NumPy array back to nested list

   Calling arr.tolist() converts the 2D array back to the original nested Python list structure.

3. Final Answer:

   arr = np.array(nested) new_list = arr.tolist() -> Option B

4. Quick Check:

   np.array() + tolist() preserves nested lists [OK]

Hint: Use np.array() then tolist() for nested list conversion [OK]

Common Mistakes:

• Using list(arr) which flattens the array
• Calling arr.to_list() which is invalid
• Trying to convert list before np.array()