NumPy Cheat Sheet for Python Data Science
NumPy Cheat Sheet for Python Data Science
>>>
>>> a.shape #Array
a.shape #Array dimensions
dimensions >>>
>>> a.sort()
a.sort() #Sort
#Sort an
an array
array
>>>
>>> len(a)
Llen(a) #Length
#Length ofof array
array
>>> c.sort(axis=0) #Sort the elements of an array's axis
>>>
>>> b.ndim
b.ndim # Number of
#Number of array
array dimensions
dimensions
elements
>>> b.dtype
b.dtype # Data type
type of
of array
array elements
*
>>>
>>> b.dtype.name
b.dtype.name # Name of
#Name of data
data type
type
>>> b.astype(int) #Convert an array to a different type
Learn
Learn NumPy
NumPy online
online at
at www.DataCamp.com
www.DataCamp.com Subsetting
Subsetting
>>>
>>> b
[2] #Select
al[2] #Select the
[1,2] #Select
b[1,2]
the element
element at
#Select the
at the
the element
element at
the 2nd
at row 1
row
index
2nd index
1 column
column 2 equivalent to
2 ((equivalent to b[1][2])
b[1][2])
1
1.5 2
2 3
3
6.
6.0 0 4 5 6
>>>
>>> np.int64
np.int64 #Signed
#Signed 64-bit
64-bit integer
integer types
types
Numpy
>>>
>>> np.float32
np.float32 #Standard
#Standard double-precision floating point
double-precision floating point
Slicin
Slicingg
>>>
>>> np.complex
np.complex #Complex
#Complex numbers
numbers represented
represented by
by 128 floats
1 1 2 3
>>>
>>> Numpy
np.bool
np.bool #Boolean
#Boolean type
type storing
storing TRUE
TRUE and
and FALSE
FALSE values
2])
>>>
>>> np.object
np.object #Python
#Python object
object type
type >>>
>>> b[0:2,1] #Select
b[0:2,1] #Select items
items atat ro ws 0
rows 0 and
and 1 in column
1 in column 1
1 1.5 2 3
>>> np.string_
np.string_ #Fixed-length
#Fixed-length string
string type
The
The NumPy
NumPy lilibrary
brary is
is the
the core lilibrary
core brary ffor
or sscientific
cientific c omputing iin
computing n
Python
Python..
>>>
>>>
type
np.unicode_ #Fixed-length unicode type
array(
array([[ 2.,
2., 5 .])
5.]) 4 5 6
>>>
>>> b[:1] #Select
#Select allall items
items at
at ro w 0 equivalent to
0 ((equivalent to b[0:1, :])
3.]])
1.5 2 3
4 5 6
w orking w
working ith these
with these a rrays
arrays >>>
>>> c[1,...] #Same
c[1,...] #Same asas [ 1,:,:]
[1,:,:]
1.],
Use the
Use ollowing import
the ffollowing import convention:
convention: [[ 4.,
4., 5 ., 6.
5., ]]])
6.]]])
>>> a[[ :: ::-1]
>>> a -1] #Reversed
#Reversed arrayarray a a array ([3, 2
array([3, 2,, 1])
1])
>>>
>>> import
import numpy
numpy as
as np
np
BBoolean
oolean IIndexing
ndexing
Ar ithmetic O
Arithmetic perations
Operations >>>
>>> a[a<2] #Select
ala<2] #Select elements from aa less
elements from less than
than 2
2 1 2 3
NumPy
NumPy Arrays
Arrays array([1])
array([1])
>>>
>>> g
g =
= aa - bb #Subtraction
>>>
>>> bb[[1,
[[1, 00,, 11,, 0] ,[0, 11,, 2,
0],[0, 0]] #Select
2, 0]] #Select elements 1,0),(
elements ((1,0), 0,1),(1,2) and
(0,1),(1,2) 0,0)
and ((0,0)
1D array 2D array [- 3. ,, -3.
[-3. 3. ,, -3.3. ]]]]))
array(
array([[ 4.
4. ,, 2.2. ,, 6.
6. ,, 1 .5])
1.5])
>>>
>>> np.subtract(a,b)
np.subtract(a,b) #Subtraction
#Subtraction
>>>
>>> b [[1, 0
b[[1, 0,, 11,, 0]][ :,[0,1,2,0]] #S#Select
OJIJ[:,[0,1,2,0]] elect aa subset
subset of
of the matrix
the ’s ro
matrix’s ws and
rows and column s
columns
>>>
>>> b + a
b + a #Addition
#Addition
array( [[ 4.
array([[ 4. ,,5.
5. ,, 6.
6. ,, 4.
4. ]J,,
array([[ 2.
array([[ 5, 4.
2.5, 4. ,, 6.
6. ]],,
[[ 1.5, 2.
1.5, 2. ,, 3.
3. ,, 1 .5],
1.5],
[ 5. , 7. , 9. ]]
[5.,7.,9. ]] )
[ 4. ,, 55.. ,, 6.
[4. 6 ,, 4.4. ]],,
>>>
>>> np.add(b,a)
np.add(b,a) Addition
Addition
[ 1.5, 2. , 3. , 1.5]])
>>>
>>> a
a / bb #Division
#Division
array([[ 0
array([[ .66666667, 11.. ,, 11.. ]J,,
0.66666667,
0.25 ,, 0
[[0.25 .4 , 00.5
0.4, .5 ]]]]))
>>>
>>> np.divide(a,b)
>>>
np.divide(a,b) #
>>> a
a * bb #
Division
#Division
Multiplication
[ 4. ,, 10
[4. 10.. ,, 18 . ]]
18. ]]))
>>>
>>> np.multiply(a,b)
np.muLltiply(a,b) # Multiplication
#MulLtiplication TTransposing
ransposing A rray
Array
>>>
>>> np.exp(b)
np.exp(b) #Exponentiation
#Exponentiation
>>> i == np.transpose(b)
>>> i np.transpose(b) #Permute
#Permute array
array dimensions
dimensions
>>>
>>> aa =
= np.array([1,2,3])
np.array([1,2,3]) >>>
>>> np.sqrt(b) #S
np.sqrt(b) quare root
float) >>>
>>> np.sin(a)
np.sin(a) #Print
#Print sines
sines of of an
an array
array
>>>
>>> cc =
= np.array([[(1.5,2,3), (4,
np.array([[(1.5,2,3), 5,6)],[(3,2,1), (4,
(4,5,6)],[(3,2,1), 5,6)]], dtype
(4,5,6)]], dtype =
= float)
float) >>>
>>> np.cos(b)
np.cos(b) #Element- wise cosine
array
>>>
>>> e.dot(f)
e.dot(f) # Dot product
#Dot product
>>>
>>> g.res #Reshape,, but
hape(3, 2) #Reshape
g.reshape(3,-2) but don’t change
don’t change data
data
nitial Pla
IInitial ceholders
Placeholders array([[ 7
array([[
[[ 7
., 7
7.,
., 7
7.,
.],
7.],
.]])
7.]]) Adding/Removing EElements
Adding/Removing lements
>>>
>>> h .resize((2,6)) #Return
h.resize((2,6)) #Return a a ne
neww array
array w ith shape
with 2,6)
shape ((2,6)
>>>
>>> np. zeros((3,4)) #Create
np.zeros((3,4)) #Create an
an array
array of
of z eros
zeros >>>
>>> np.append( h,g) #Append
#Append items
items to to an
an array
Comparison
np.append(h,g) array
>>>
>>> np.ones((2,3,4),dtype=np.int
np.ones((2,3,4),dtype=np.int16) 16) #Create
#Create an
an array
array of
of ones
array
>>>
>>> d
d =
= np.arange( 10,25,5) #Create
np.arange(10,25,5) #Create an
an array
array of
of evenly
evenly spaced
spaced values step value
values ((step value))
>>>
>>> np.delete(a, [1]) #
np.delete(a,[1]) Delete items
#Delete from an
items from an array
array
>>>
>>> np.linspace( 0,2,9) #Create
np. linspace(0,2,9) #Create anan array
array of
of evenly
evenly spaced
spaced values number of
values ((number of samples
samples))
>>>
>>> e
e =
= np.full((2,2), 7) #Create
np.full((2,2),7) #Create aa constant
constant array
array >>>
>>> a = b
a == b #Element- wise comparison
arrays
>>>
>>> np.random.random((2,2))
np.random.random((2,2)) #Create
#Create an
an array
array w ith random
random values
False, F
[[False, alse, FFalse]],
False, alse]], dtype=bool)
dtype=bool)
with values array(
array([[ 11,, 2,
2, 3,
3, 1010,, 15
15,, 2 0])
20])
>>>
>>> np.empty((3,2))
np.empty((3,2)) #Create
#Create an
an empty
empty array
array >>>
>>> a < 2
a < 2 #Element- wise comparison
#Element-wise comparison
>>>
>>> np.vstac k((a,b)) #Stac
np.vstack((a,b)) #Stack k arrays
arrays vertically row wise)
vertically ((row-wise)
array([True, F
array([True, alse, FFalse],
False, alse], dtype=bool)
dtype=bool) array([[ 1
array([[ 1.. ,, 2.
2. ,, 3.
3. ]],,
>>>
>>> np.array_e qual(a, b)
np.array_equal(a, b) #Array- wise comparison
#Array-wise comparison [[ 1 .5, 2.
1.5, 2. ,, 3.
3. ]],,
[[ 4.
4. , 5. , 6. ]] )
> I/O
,5., 6 ]]
>>>
>>> np.r_ [e,f] #Stac
np.r_Le,f] #Stack k arrays
arrays vertically row wise)
vertically ((row-wise)
Aggregate Fun
Aggregate ctions
Functions >>>
>>> np.hstack((e,f)) #Stac
np.hstack((e,f)) #Stack k arrays
arrays hori zontally ((column-wise)
horizontally column-wise)
array([[ 7
array([[ ., 77.,
7., ., 1 ., 00.],
1., .],
., 77.,
., 0@.,
., 1 .]])
a.sum() >>>
>>> np.column_stac k((a,d)) #Create
np.coLumn_stack((a,d)) #Create stac ked column-
stacked wise arrays
column-wise arrays
>>>
>>> a.min() #Array-wise minimum
a.min() value
>>>
>>> b.max(axis=0) #Maximum value
b.max(axis=0) value of an array row
[[ 2, 15],,
2, 15]
>>>
>>> np.save('my_array', a)
np.save('my_array', a) >>>
>>> b.cumsum(axis=1) #Cumulative sum
b.cumsum(axis=1) sum of the elements
[[ 3,
3, 2 0]])
20]])
>>>
>>> np.savez( 'array.npz', a,
np.savez('array.npz', a, b)
b) >>>
>>> a.mean()
a.mean() #Mean
>>>
>>> np.c_ [a,d] #Create
np.c_La,d] #Create stac
stackedked column- wise arrays
column-wise arrays
>>>
>>> np.load('my_array.npy')
np. Load('my_array.npy') >>>
>>> np.median(b) #Median
np.median(b) #Median
>>>
>>> np.corrcoef(a) #Correlation coefficient
np.corrcoef(a) Splittnp.vsplit(c,2)
ing A
Splitting
>>>
array([[[
[array([[[ rray
Arrayss1.5,2.,
[array([1]),array([2]),array([3])]
np.hsplit(a,3)
[ [4.
3.,
4., 5.
, 2.
5. #Split
, 6.
1. ]]]),
],the array horizontally
vertically at at
thethe2nd
3rd
index
index
>>>
>>> np.std(b) #Standard deviation
np.std(b) >>> np.hsplit(a,3) #Split the array horizontally at the 3rd index
Saving
saving
>>>
np. &
& Loading
Loading Text Files
np.genfromtxt("my_file.csv",
np.savetxt("myarray.txt",
lLoadtxt("myfile.txt")
Text
a, delimiter="
delimiter=',
Files ')") [array([1]),array([2]),array([3])]
>>> np.loadtxt("myfile.txt")
[ 4. , 5. , 6. ]]]),
>>>
>>> hh =
= a.view() #Create
a.view() #Create a
a view of
view of the
the array
array with the
with the same
same data
data
>>>
>>> np.copy(a)
np.copy(a) #Create
#Create a
a copy
copy of
of the
the array
array