python-numpyHow to split Numpy array by chunk length
import math
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8])
chunk_size = 2
split = np.array_split(a, math.ceil(a.size/chunk_size))ctrl + cimport numpy as npload Numpy module for Python | np.array(declare Numpy array |
np.array_splitsplit array into specified number of sub-arrays; returns a list of sub-arrays | chunk_sizenumber of elements in each chunk (chunk size) after split |
math.ceil(a.size/chunk_size)calculate number of chunks to split by based on chunk size and given array size | |
Usage example
import math
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8])
chunk_size = 2
split = np.array_split(a, math.ceil(a.size/chunk_size))
print(split)output
[array([1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]
Related
More of Python Numpy
- How to use Numpy binary search
- How to get last column from Numpy array
- How to add column to Numpy array
- How to save Numpy array to CSV
- Save Numpy array to image
- How to get real part from complex number
- How to get specific column from Numpy array
- Import image data to Numpy array
- How to use where() to filter array
- Numpy histogram example
See more codes...