np.take() and np.put() for advanced selection in NumPy - Cheat Sheet & Quick Revision

Recall & Review

beginner

What does np.take() do in NumPy?

np.take() selects elements from an array at specified indices. It is like picking items from a list using a list of positions.

beginner

How does np.put() work in NumPy?

np.put() replaces elements in an array at given indices with new values. Think of it as putting new items into specific spots in a list.

intermediate

What is the difference between np.take() and simple indexing like arr[indices]?

np.take() can handle multi-dimensional arrays with an axis argument, allowing selection along a specific axis. Simple indexing selects elements but may not support advanced axis selection.

intermediate

Can np.put() modify elements along a specific axis?

No, np.put() works on the flattened version of the array by default. To modify along an axis, you need to reshape or use other methods.

advanced

What happens if you use np.put() with indices outside the array size?

Indices are taken modulo the array size, so they wrap around. This means out-of-range indices loop back to valid positions.

What does np.take(arr, indices) return?

A Elements from <code>arr</code> at positions in <code>indices</code>

B A new array with elements replaced at <code>indices</code>

C The sum of elements at <code>indices</code>

D The shape of <code>arr</code>

np.take() picks elements from the array at the specified indices.

Which function would you use to replace elements in an array at specific positions?

A np.take()

B np.put()

C np.sum()

D np.reshape()

np.put() replaces elements at given indices.

What does np.put(arr, [0, 2], [10, 20]) do?

A Selects elements at positions 0 and 2

B Deletes elements at positions 0 and 2

C Returns the sum of elements at positions 0 and 2

D Replaces elements at positions 0 and 2 with 10 and 20

np.put() replaces elements at specified indices with new values.

If you want to select elements along a specific axis, which function is better?

A np.put()

B np.flatten()

C np.take() with axis argument

D np.sum()

np.take() supports an axis argument for selection along that axis.

What happens if np.put() receives an index larger than the array size?

A Wraps around using modulo

B Raises an error

C Ignores the index

D Deletes the last element

Indices wrap around using modulo the array size.

Explain how np.take() can be used to select elements from a 2D array along a specific axis.

Think about how you pick rows or columns by specifying the axis.

Describe the behavior of np.put() when given indices and values, including what happens with out-of-range indices.

Consider how the function modifies the array and handles indices beyond its size.

Practice

1. What does np.take() do in NumPy?

easy

A. Selects elements from an array using specified indices

B. Changes elements in an array at specified indices

C. Sorts the elements of an array

D. Creates a new array filled with zeros

Solution

1. Step 1: Understand the function purpose

   np.take() is used to pick elements from an array by their index positions.

2. Step 2: Compare with other options

   Only Selects elements from an array using specified indices describes selecting elements by indices, while others describe different functions.

3. Final Answer:

   Selects elements from an array using specified indices -> Option A

4. Quick Check:

   np.take() = select elements [OK]

Hint: np.take() picks by index, np.put() changes by index [OK]

Common Mistakes:

• Confusing np.take() with np.put()
• Thinking np.take() modifies the array
• Mixing np.take() with sorting functions

2. Which of the following is the correct syntax to replace elements at indices [1, 3] in array a with values [10, 20] using np.put()?

easy

A. np.put(a, [1, 3], [10, 20])

B. np.put([1, 3], a, [10, 20])

C. np.put(a, [10, 20], [1, 3])

D. np.put(a, 10, [1, 3, 20])

Solution

1. Step 1: Recall np.put() syntax

   The correct syntax is np.put(array, indices, values) to replace elements at given indices.

2. Step 2: Match syntax with options

   np.put(a, [1, 3], [10, 20]) matches the correct order: array, indices, values.

3. Final Answer:

   np.put(a, [1, 3], [10, 20]) -> Option A

4. Quick Check:

   np.put(array, indices, values) [OK]

Hint: np.put(array, indices, values) replaces elements [OK]

Common Mistakes:

• Swapping the order of arguments
• Passing values as indices
• Using wrong data types for indices or values

3. What is the output of the following code?

import numpy as np
a = np.array([5, 10, 15, 20, 25])
indices = [0, 2, 4]
result = np.take(a, indices)
print(result)

medium

A. [10 20 25]

B. [5 15 25]

C. [5 10 15]

D. [0 2 4]

Solution

1. Step 1: Understand np.take() with given indices

   np.take(a, [0, 2, 4]) selects elements at positions 0, 2, and 4 from array a.

2. Step 2: Extract elements from array

   Elements at these positions are 5 (index 0), 15 (index 2), and 25 (index 4).

3. Final Answer:

   [5 15 25] -> Option B

4. Quick Check:

   np.take() picks elements by index [OK]

Hint: np.take() returns elements at given indices [OK]

Common Mistakes:

• Confusing indices with values
• Expecting output as original array
• Mixing up index positions

4. Identify the error in this code snippet:

import numpy as np
a = np.array([1, 2, 3, 4])
np.put(a, [1, 4], [10, 20])
print(a)

medium

A. Syntax error in np.put() usage

B. Values list length does not match indices length

C. Index 4 is out of bounds for array of size 4

D. np.put() cannot modify arrays in-place

Solution

1. Step 1: Check array size and indices

   Array a has size 4, valid indices are 0 to 3. Index 4 is invalid.

2. Step 2: Understand np.put() behavior

   np.put() tries to assign values at given indices. Using index 4 causes an IndexError.

3. Final Answer:

   Index 4 is out of bounds for array of size 4 -> Option C

4. Quick Check:

   Index error due to invalid index [OK]

Hint: Check indices are within array size before np.put() [OK]

Common Mistakes:

• Ignoring array size limits
• Assuming np.put() auto-expands array
• Confusing index and value positions

5. Given a = np.array([1, 2, 3, 4, 5]), how can you use np.put() and np.take() together to replace elements at indices [0, 2] with [10, 20] and then retrieve these updated elements?

hard

A. np.take(a, [10, 20]); np.put(a, [0, 2])

B. np.take(a, [0, 2], [10, 20]); np.put(a, [0, 2])

C. np.put([0, 2], a, [10, 20]); np.take([0, 2], a)

D. np.put(a, [0, 2], [10, 20]); np.take(a, [0, 2])

Solution

1. Step 1: Use np.put() to update elements

   np.put(a, [0, 2], [10, 20]) replaces elements at indices 0 and 2 with 10 and 20.

2. Step 2: Use np.take() to retrieve updated elements

   np.take(a, [0, 2]) returns the updated elements at those indices.

3. Final Answer:

   np.put(a, [0, 2], [10, 20]); np.take(a, [0, 2]) -> Option D

4. Quick Check:

   Put updates, take retrieves updated values [OK]

Hint: Put first to update, then take to get new values [OK]

Common Mistakes:

• Mixing argument order in np.put()
• Using np.take() to update values
• Confusing arrays and indices in functions