2016-07-16 15:42:18 +02:00
2019-11-23 21:05:14 +01:00
# 100 numpy exercises
2016-07-16 15:42:18 +02:00
2019-11-23 21:05:14 +01:00
This is a collection of exercises that have been collected in the numpy mailing list , on stack overflow
and in the numpy documentation . The goal of this collection is to offer a quick reference for both old
and new users but also to provide a set of exercises for those who teach .
2016-07-16 15:42:18 +02:00
2019-11-23 21:05:14 +01:00
If you find an error or think you ' ve a better way to solve some of them, feel
free to open an issue at < https : / / github . com / rougier / numpy - 100 >
File automatically generated . See the documentation to update questions / answers / hints programmatically .
#### 1. Import the numpy package under the name `np` (★☆☆)
Hint : ` hint : import … as `
2016-07-16 15:42:18 +02:00
#### 2. Print the numpy version and the configuration (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . __version__ , np . show_config ) `
2016-07-16 15:42:18 +02:00
#### 3. Create a null vector of size 10 (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . zeros `
#### 4. How to find the memory size of any array (★☆☆)
Hint : ` hint : size , itemsize `
#### 5. How to get the documentation of the numpy add function from the command line? (★☆☆)
Hint : ` hint : np . info `
#### 6. Create a null vector of size 10 but the fifth value which is 1 (★☆☆)
Hint : ` hint : array [ 4 ] `
#### 7. Create a vector with values ranging from 10 to 49 (★☆☆)
Hint : ` hint : arange `
#### 8. Reverse a vector (first element becomes last) (★☆☆)
Hint : ` hint : array [ : : - 1 ] `
#### 9. Create a 3x3 matrix with values ranging from 0 to 8 (★☆☆)
Hint : ` hint : reshape `
#### 10. Find indices of non-zero elements from [1,2,0,0,4,0] (★☆☆)
Hint : ` hint : np . nonzero `
2016-07-16 15:42:18 +02:00
#### 11. Create a 3x3 identity matrix (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . eye `
2016-07-16 15:42:18 +02:00
#### 12. Create a 3x3x3 array with random values (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . random . random `
2016-07-16 15:42:18 +02:00
#### 13. Create a 10x10 array with random values and find the minimum and maximum values (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : min , max `
2016-07-16 15:42:18 +02:00
#### 14. Create a random vector of size 30 and find the mean value (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : mean `
2016-07-16 15:42:18 +02:00
#### 15. Create a 2d array with 1 on the border and 0 inside (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : array [ 1 : - 1 , 1 : - 1 ] `
2016-07-16 15:42:18 +02:00
#### 16. How to add a border (filled with 0's) around an existing array? (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . pad `
2016-07-16 15:42:18 +02:00
#### 17. What is the result of the following expression? (★☆☆)
` ` ` python
0 * np . nan
np . nan == np . nan
np . inf > np . nan
np . nan - np . nan
2018-10-31 09:42:02 +01:00
np . nan in set ( [ np . nan ] )
2016-07-16 15:42:18 +02:00
0.3 == 3 * 0.1
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` hint : NaN = not a number , inf = infinity `
2016-07-16 15:42:18 +02:00
#### 18. Create a 5x5 matrix with values 1,2,3,4 just below the diagonal (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . diag `
2016-07-16 15:42:18 +02:00
#### 19. Create a 8x8 matrix and fill it with a checkerboard pattern (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : array [ : : 2 ] `
2016-07-16 15:42:18 +02:00
#### 20. Consider a (6,7,8) shape array, what is the index (x,y,z) of the 100th element?
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . unravel_index `
2016-07-16 15:42:18 +02:00
#### 21. Create a checkerboard 8x8 matrix using the tile function (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . tile `
2016-07-16 15:42:18 +02:00
#### 22. Normalize a 5x5 random matrix (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : ( x - mean ) / std `
2017-03-30 18:35:50 +02:00
#### 23. Create a custom dtype that describes a color as four unsigned bytes (RGBA) (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . dtype `
2016-07-16 15:42:18 +02:00
#### 24. Multiply a 5x3 matrix by a 3x2 matrix (real matrix product) (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : `
2016-07-16 15:42:18 +02:00
#### 25. Given a 1D array, negate all elements which are between 3 and 8, in place. (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : > , < = `
2016-07-16 15:42:18 +02:00
#### 26. What is the output of the following script? (★☆☆)
` ` ` python
# Author: Jake VanderPlas
print ( sum ( range ( 5 ) , - 1 ) )
from numpy import *
print ( sum ( range ( 5 ) , - 1 ) )
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . sum `
2016-07-16 15:42:18 +02:00
#### 27. Consider an integer vector Z, which of these expressions are legal? (★☆☆)
` ` ` python
Z * * Z
2 << Z >> 2
Z < - Z
1 j * Z
Z / 1 / 1
Z < Z > Z
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` No hints provided . . . `
2016-07-16 15:42:18 +02:00
#### 28. What are the result of the following expressions?
` ` ` python
np . array ( 0 ) / np . array ( 0 )
np . array ( 0 ) / / np . array ( 0 )
np . array ( [ np . nan ] ) . astype ( int ) . astype ( float )
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` No hints provided . . . `
2016-07-16 15:42:18 +02:00
#### 29. How to round away from zero a float array ? (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . uniform , np . copysign , np . ceil , np . abs `
2016-07-16 15:42:18 +02:00
#### 30. How to find common values between two arrays? (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . intersect1d `
2016-07-16 15:42:18 +02:00
#### 31. How to ignore all numpy warnings (not recommended)? (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . seterr , np . errstate `
2016-07-16 15:42:18 +02:00
#### 32. Is the following expressions true? (★☆☆)
` ` ` python
np . sqrt ( - 1 ) == np . emath . sqrt ( - 1 )
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` hint : imaginary number `
2016-07-16 15:42:18 +02:00
#### 33. How to get the dates of yesterday, today and tomorrow? (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . datetime64 , np . timedelta64 `
2016-07-16 15:42:18 +02:00
#### 34. How to get all the dates corresponding to the month of July 2016? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . arange ( dtype = datetime64 [ ' D ' ] ) `
#### 35. How to compute ((A+B)*(-A/2)) in place (without copy)? (★★☆)
Hint : ` hint : np . add ( out = ) , np . negative ( out = ) , np . multiply ( out = ) , np . divide ( out = ) `
2016-07-16 15:42:18 +02:00
#### 36. Extract the integer part of a random array using 5 different methods (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : % , np . floor , np . ceil , astype , np . trunc `
2016-07-16 15:42:18 +02:00
#### 37. Create a 5x5 matrix with row values ranging from 0 to 4 (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . arange `
2016-07-16 15:42:18 +02:00
#### 38. Consider a generator function that generates 10 integers and use it to build an array (★☆☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . fromiter `
2016-07-16 15:42:18 +02:00
#### 39. Create a vector of size 10 with values ranging from 0 to 1, both excluded (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . linspace `
2016-07-16 15:42:18 +02:00
#### 40. Create a random vector of size 10 and sort it (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : sort `
2016-07-16 15:42:18 +02:00
#### 41. How to sum a small array faster than np.sum? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . add . reduce `
2017-03-30 18:35:50 +02:00
#### 42. Consider two random array A and B, check if they are equal (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . allclose , np . array_equal `
2016-07-16 15:42:18 +02:00
#### 43. Make an array immutable (read-only) (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : flags . writeable `
2016-07-16 15:42:18 +02:00
#### 44. Consider a random 10x2 matrix representing cartesian coordinates, convert them to polar coordinates (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . sqrt , np . arctan2 `
#### 45. Create random vector of size 10 and replace the maximum value by 0 (★★☆)
Hint : ` hint : argmax `
#### 46. Create a structured array with `x` and `y` coordinates covering the [0,1]x[0,1] area (★★☆)
Hint : ` hint : np . meshgrid `
#### 47. Given two arrays, X and Y, construct the Cauchy matrix C (Cij =1/(xi - yj))
Hint : ` hint : np . subtract . outer `
2016-07-16 15:42:18 +02:00
#### 48. Print the minimum and maximum representable value for each numpy scalar type (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . iinfo , np . finfo , eps `
2016-07-16 15:42:18 +02:00
#### 49. How to print all the values of an array? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . set_printoptions `
2017-05-25 17:25:13 +02:00
#### 50. How to find the closest value (to a given scalar) in a vector? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : argmin `
2016-07-16 15:42:18 +02:00
#### 51. Create a structured array representing a position (x,y) and a color (r,g,b) (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : dtype `
2016-07-16 15:42:18 +02:00
#### 52. Consider a random vector with shape (100,2) representing coordinates, find point by point distances (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . atleast_2d , T , np . sqrt `
2016-07-16 15:42:18 +02:00
#### 53. How to convert a float (32 bits) array into an integer (32 bits) in place?
2019-11-23 21:05:14 +01:00
Hint : ` hint : astype ( copy = False ) `
2016-07-16 15:42:18 +02:00
#### 54. How to read the following file? (★★☆)
` ` `
1 , 2 , 3 , 4 , 5
6 , , , 7 , 8
, , 9 , 10 , 11
` ` `
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . genfromtxt `
2016-07-16 15:42:18 +02:00
#### 55. What is the equivalent of enumerate for numpy arrays? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . ndenumerate , np . ndindex `
2016-07-16 15:42:18 +02:00
#### 56. Generate a generic 2D Gaussian-like array (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . meshgrid , np . exp `
2016-07-16 15:42:18 +02:00
#### 57. How to randomly place p elements in a 2D array? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . put , np . random . choice `
2016-07-16 15:42:18 +02:00
#### 58. Subtract the mean of each row of a matrix (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : mean ( axis = , keepdims = ) `
2016-07-16 15:42:18 +02:00
#### 59. How to sort an array by the nth column? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : argsort `
2016-07-16 15:42:18 +02:00
#### 60. How to tell if a given 2D array has null columns? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : any , ~ `
2016-07-16 15:42:18 +02:00
#### 61. Find the nearest value from a given value in an array (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . abs , argmin , flat `
2016-07-16 15:42:18 +02:00
#### 62. Considering two arrays with shape (1,3) and (3,1), how to compute their sum using an iterator? (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . nditer `
2016-07-16 15:42:18 +02:00
#### 63. Create an array class that has a name attribute (★★☆)
2019-11-23 21:05:14 +01:00
Hint : ` hint : class method `
2016-07-16 15:42:18 +02:00
#### 64. Consider a given vector, how to add 1 to each element indexed by a second vector (be careful with repeated indices)? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . bincount | np . add . at `
2016-07-16 15:42:18 +02:00
#### 65. How to accumulate elements of a vector (X) to an array (F) based on an index list (I)? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . bincount `
2016-07-16 15:42:18 +02:00
#### 66. Considering a (w,h,3) image of (dtype=ubyte), compute the number of unique colors (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . unique `
2016-07-16 15:42:18 +02:00
#### 67. Considering a four dimensions array, how to get sum over the last two axis at once? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : sum ( axis = ( - 2 , - 1 ) ) `
2016-07-16 15:42:18 +02:00
#### 68. Considering a one-dimensional vector D, how to compute means of subsets of D using a vector S of same size describing subset indices? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . bincount `
2016-07-16 15:42:18 +02:00
#### 69. How to get the diagonal of a dot product? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . diag `
#### 70. Consider the vector [1, 2, 3, 4, 5], how to build a new vector with 3 consecutive zeros interleaved between each value? (★★★)
Hint : ` hint : array [ : : 4 ] `
2016-07-16 15:42:18 +02:00
#### 71. Consider an array of dimension (5,5,3), how to mulitply it by an array with dimensions (5,5)? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : array [ : , : , None ] `
2016-07-16 15:42:18 +02:00
#### 72. How to swap two rows of an array? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : array [ [ ] ] = array [ [ ] ] `
2016-07-16 15:42:18 +02:00
#### 73. Consider a set of 10 triplets describing 10 triangles (with shared vertices), find the set of unique line segments composing all the triangles (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : repeat , np . roll , np . sort , view , np . unique `
2016-07-16 15:42:18 +02:00
#### 74. Given an array C that is a bincount, how to produce an array A such that np.bincount(A) == C? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . repeat `
2016-07-16 15:42:18 +02:00
#### 75. How to compute averages using a sliding window over an array? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . cumsum `
#### 76. Consider a one-dimensional array Z, build a two-dimensional array whose first row is (Z[0],Z[1],Z[2]) and each subsequent row is shifted by 1 (last row should be (Z[-3],Z[-2],Z[-1]) (★★★)
Hint : ` hint : from numpy . lib import stride_tricks `
2016-07-16 15:42:18 +02:00
#### 77. How to negate a boolean, or to change the sign of a float inplace? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . logical_not , np . negative `
#### 78. Consider 2 sets of points P0,P1 describing lines (2d) and a point p, how to compute distance from p to each line i (P0[i],P1[i])? (★★★)
Hint : ` No hints provided . . . `
#### 79. Consider 2 sets of points P0,P1 describing lines (2d) and a set of points P, how to compute distance from each point j (P[j]) to each line i (P0[i],P1[i])? (★★★)
Hint : ` No hints provided . . . `
2016-07-16 15:42:18 +02:00
#### 80. Consider an arbitrary array, write a function that extract a subpart with a fixed shape and centered on a given element (pad with a `fill` value when necessary) (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : minimum maximum `
#### 81. Consider an array Z = [1,2,3,4,5,6,7,8,9,10,11,12,13,14], how to generate an array R = [[1,2,3,4], [2,3,4,5], [3,4,5,6], ..., [11,12,13,14]]? (★★★)
Hint : ` hint : stride_tricks . as_strided `
#### 82. Compute a matrix rank (★★★)
Hint : ` hint : np . linalg . svd `
2016-07-16 15:42:18 +02:00
#### 83. How to find the most frequent value in an array?
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . bincount , argmax `
2016-07-16 15:42:18 +02:00
#### 84. Extract all the contiguous 3x3 blocks from a random 10x10 matrix (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : stride_tricks . as_strided `
#### 85. Create a 2D array subclass such that Z[i,j] == Z[j,i] (★★★)
Hint : ` hint : class method `
2016-07-16 15:42:18 +02:00
#### 86. Consider a set of p matrices wich shape (n,n) and a set of p vectors with shape (n,1). How to compute the sum of of the p matrix products at once? (result has shape (n,1)) (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . tensordot `
2016-07-16 15:42:18 +02:00
#### 87. Consider a 16x16 array, how to get the block-sum (block size is 4x4)? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . add . reduceat `
2016-07-16 15:42:18 +02:00
#### 88. How to implement the Game of Life using numpy arrays? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` No hints provided . . . `
2016-07-16 15:42:18 +02:00
#### 89. How to get the n largest values of an array (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . argsort | np . argpartition `
2016-07-16 15:42:18 +02:00
#### 90. Given an arbitrary number of vectors, build the cartesian product (every combinations of every item) (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . indices `
2016-07-16 15:42:18 +02:00
#### 91. How to create a record array from a regular array? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . core . records . fromarrays `
2016-07-16 15:42:18 +02:00
#### 92. Consider a large vector Z, compute Z to the power of 3 using 3 different methods (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . power , * , np . einsum `
2016-07-16 15:42:18 +02:00
#### 93. Consider two arrays A and B of shape (8,3) and (2,2). How to find rows of A that contain elements of each row of B regardless of the order of the elements in B? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . where `
#### 94. Considering a 10x3 matrix, extract rows with unequal values (e.g. [2,2,3]) (★★★)
Hint : ` No hints provided . . . `
2016-07-16 15:42:18 +02:00
#### 95. Convert a vector of ints into a matrix binary representation (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . unpackbits `
2016-07-16 15:42:18 +02:00
#### 96. Given a two dimensional array, how to extract unique rows? (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . ascontiguousarray | np . unique `
2016-07-16 15:42:18 +02:00
#### 97. Considering 2 vectors A & B, write the einsum equivalent of inner, outer, sum, and mul function (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . einsum `
2016-07-16 15:42:18 +02:00
#### 98. Considering a path described by two vectors (X,Y), how to sample it using equidistant samples (★★★)?
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . cumsum , np . interp `
2016-07-16 15:42:18 +02:00
#### 99. Given an integer n and a 2D array X, select from X the rows which can be interpreted as draws from a multinomial distribution with n degrees, i.e., the rows which only contain integers and which sum to n. (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . logical_and . reduce , np . mod `
2016-07-16 15:42:18 +02:00
#### 100. Compute bootstrapped 95% confidence intervals for the mean of a 1D array X (i.e., resample the elements of an array with replacement N times, compute the mean of each sample, and then compute percentiles over the means). (★★★)
2019-11-23 21:05:14 +01:00
Hint : ` hint : np . percentile `