Numpy Cheatsheet
Numpy Cheatsheet
IMPORTING NUMPY
import numpy as np
CREATING ARRAYS
np.array([1, 2, 3])
np.zeros((2,3))
np.ones((3,2))
np.eye(3) # Identity matrix
np.arange(0, 10, 2)
np.linspace(0, 1, 5)
np.random.rand(2,3) # Random 2x3 array
ARRAY PROPERTIES
a.shape # Dimensions
a.size # Total elements
a.dtype # Data type
RESHAPING / TRANSFORMING
a.reshape((3,2))
a.ravel() # Flatten
a.T # Transpose
INDEXING / SLICING
a[0] # First element
a[1:3] # Slice
a[1][0] or a[1, 0] # Element access
ARITHMETIC OPERATIONS
a+b
a-b
a*2
np.dot(a, b) # Matrix multiplication
AGGREGATIONS
a.sum(), a.min(), a.max()
a.mean(), a.std()
CONDITIONALS
a[a > 2] # Filter elements
np.where(a > 2, 1, 0) # Replace based on condition
SAVING / LOADING
np.save("data.npy", a)
b = np.load("data.npy")