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 save Numpy array to CSV
- How to sort Numpy array by column
- How to join 2 Numpy arrays
- Numpy histogram example
- Save Numpy array to image
- How to use Numpy bincount to find value occurrences
- How to use Numpy binary search
- How to load Numpy matrix from text file
- Import image data to Numpy array
- How to use zip for Numpy arrays
See more codes...