NumPy-3
NumPy Splitting Array
Splitting is reverse operation of Joining.
Joining merges multiple arrays into one and Splitting breaks one array into multiple.
We use array_split() for splitting arrays, we pass it the array we want to split and the number of splits.
Example
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
newarr = np.array_split(arr, 3)
print(newarr)
>>> [array([1, 2]), array([3, 4]), array([5, 6])]
Note
"We also have the method split() available but it will not adjust the elements when elements are less in source array for splitting like in example above, array_split() worked properly but split() would fail."
The return value of the array_split() method is an array containing each of the split as an array.
Example
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
newarr = np.array_split(arr, 3)
print(newarr[0])
>>> [1 2]
print(newarr[1])
>>> [3 4]
print(newarr[2])
>>> [5 6]
Splitting 2-D Arrays
Split the 2-D array into three 2-D arrays.
import numpy as np
arr = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]])
newarr = np.array_split(arr, 3)
print(newarr)
>>> [array([[1, 2],
[3, 4]]), array([[5, 6],
[7, 8]]), array([[ 9, 10],
[11, 12]])]
Split the 2-D array into three 2-D arrays along columns.
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15], [16, 17, 18]])
newarr = np.array_split(arr, 3, axis=1)
print(newarr)
>>> [array([[ 1],
[ 4],
[ 7],
[10],
[13],
[16]]), array([[ 2],
[ 5],
[ 8],
[11],
[14],
[17]]), array([[ 3],
[ 6],
[ 9],
[12],
[15],
[18]])]
NumPy Searching Arrays
You can search an array for a certain value, and return the indexes that get a match.
To search an array, use the where() method.
Example
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 4])
x = np.where(arr == 4)
print(x)
>>> (array([3, 5, 6]),)
Search Sorted
There is a method called searchsorted() which performs a binary search in the array, and returns the index where the specified value would be inserted to maintain the search order.
The searchsorted() method is assumed to be used on sorted arrays.
Example
import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7)
print(x)
>>> 1
Search From the Right Side
By default the left most index is returned, but we can give side='right' to return the right most index instead.
Example
import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7, side='right')
print(x)
>>> 2
The method starts the search from the right and returns the first index where the number 7 is no longer less than the next value.
Multiple Values
To search for more than one value, use an array with the specified values.
Example
import numpy as np
arr = np.array([1, 3, 5, 7])
x = np.searchsorted(arr, [2, 4, 6])
print(x)
>>> [1 2 3]
NumPy Sorting Arrays
Sorting means putting elements in a ordered sequence.
Ordered sequence is any sequence that has an order corresponding to elements, like numeric or alphabetical, ascending or descending.
The NumPy ndarray object has a function called sort(), that will sort a specified array.
Sort the Array
import numpy as np
arr = np.array([3, 2, 0, 1])
print(np.sort(arr))
>>> [0 1 2 3]
This method returns a copy of the array, leaving the original array unchanged.
Sort the array alphabetically
import numpy as np
arr = np.array(['banana', 'cherry', 'apple'])
print(np.sort(arr))
>>> ['apple' 'banana' 'cherry']
Sort a boolean array
import numpy as np
arr = np.array([True, False, True])
print(np.sort(arr))
>>> [False True True]
Sorting a 2-D Array
If you use the sort() method on a 2-D array, both arrays will be sorted
Sort a 2-D Array
import numpy as np
arr = np.array([[3, 2, 4], [5, 0, 1]])
print(np.sort(arr))
>>> [[2 3 4]
[0 1 5]]
NumPy Filter Array
Getting some elements out of an existing array and creating a new array out of them is called filtering.
In NumPy, you filter an array using a boolean index list.
A boolean index list is a list of booleans corresponding to indexes in the array.
If the value at an index is True that element is contained in the filtered array, if the value at that index is False that element is excluded from the filtered array.
Example
import numpy as np
arr = np.array([41, 42, 43, 44])
x = [True, False, True, False]
newarr = arr[x]
print(newarr)
>>> [41 43]
The new filter contains only the values where the filter array had the value True, in this case, index 0 and 2.
Creating the Filter Array
In the example above we hard-coded the True and False values, but the common use is to create a filter array based on conditions.
Create a filter array that will return the values higher than 42
import numpy as np
arr = np.array([41, 42, 43, 44])
# Create an empty list
filter_arr = []
# go through each element in arr
for element in arr:
# if the element is higher than 42, set the value to True, otherwise False:
if element > 42:
filter_arr.append(True)
else:
filter_arr.append(False)
newarr = arr[filter_arr]
print(filter_arr)
>>> [False, False, True, True]
print(newarr)
>>> [43 44]
Creating Filter Directly From Array
We can directly substitute the array instead of the iterable variable in our condition and it will work just as we expect it to.
Create a filter array that will return the values higher than 42
import numpy as np
arr = np.array([41, 42, 43, 44])
filter_arr = arr > 42
newarr = arr[filter_arr]
print(filter_arr)
>>> [False False True True]
print(newarr)
>>> [43 44]
Create a filter array that will return only even elements from the original array
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7])
filter_arr = arr % 2 == 0
newarr = arr[filter_arr]
print(filter_arr)
>>> [False True False True False True False]
print(newarr)
>>> [2 4 6]
Random Numbers in NumPy
Random number does NOT mean a different number every time. Random means something that can not be predicted logically.
Generate Random Number
NumPy offers the random module to work with random numbers.
Generate a random integer from 0 to 100
from numpy import random
x = random.randint(100)
print(x)
>>> 98
Generate Random Float
The random module's rand() method returns a random float between 0 and 1
The random module's rand() method returns a random float between 0 and 1
from numpy import random
x = random.rand()
print(x)
>>> 0.4516533963530909
Generate Random Array
The randint() method takes a size parameter where you can specify the shape of an array.
Generate a 1-D Array
from numpy import random
x=random.randint(100, size=(5))
print(x)
>>> [23 22 74 44 50]
Generate a 2-D Array
from numpy import random
x = random.randint(100, size=(3, 5))
print(x)
>>> [[80 54 19 74 65]
[26 60 69 34 25]
[50 16 53 84 90]]
Floats
The rand() method also allows you to specify the shape of the array.
Generate a 1-D Array
from numpy import random
x = random.rand(5)
print(x)
>>> [0.6313987 0.7341789 0.0873221 0.0752967 0.2127466]
Generate a 2-D Array
from numpy import random
x = random.rand(3, 5)
print(x)
>>> [[0.03379952 0.78263517 0.9834899 0.47851523 0.02948659]
[0.36284007 0.10740884 0.58485016 0.20708396 0.00969559]
[0.88232193 0.86068608 0.75548749 0.61233486 0.06325663]]
Generate Random Number From Array
The choice()
method allows you to generate a random value based on an array of values.
This method takes an array as a parameter and randomly returns one of the values.
Example
from numpy import random
x = random.choice([3, 5, 7, 9])
print(x)
>>> 5
The choice() method also allows you to return an array of values.
Add a size parameter to specify the shape of the array.
Example
from numpy import random
x = random.choice([3, 5, 7, 9], size=(3, 5))
print(x)
>>> [[9 3 5 5 7]
[7 5 3 3 9]
[7 5 9 9 7]]