numpy.partition() in Python

numpy.partition() in Python

Last Updated : 28 Dec, 2018

numpy.partition() function is used to create a partitioned copy of input array with its elements rearranged in such a way that the value of the element in k-th position is in the position it would be in a sorted array. All elements smaller than the k-th element are moved before this element and all equal or greater are moved behind it. The ordering of the elements in the two partitions is undefined.

Syntax : numpy.partition(arr, kth, axis=-1, kind='introselect', order=None) Parameters : arr : [array_like] Input array. kth : [int or sequence of ints ] Element index to partition by. axis : [int or None] Axis along which to sort. If None, the array is flattened before sorting. The default is -1, which sorts along the last axis. kind : Selection algorithm. Default is ‘introselect’. order : [str or list of str] When arr is an array with fields defined, this argument specifies which fields to compare first, second, etc. Return : [ndarray] Partitioned array of the same type and shape as arr.

Code #1 :

Python3

# Python program explaining
# partition() function
 
import numpy as geek

# input array
in_arr = geek.array([ 2, 0,  1, 5, 4, 9])
print ("Input array : ", in_arr) 

out_arr = geek.partition(in_arr, 3)
print ("Output partitioned array : ", out_arr)

Output:

Input array :  [2 0 1 5 4 9]
Output partitioned array :  [0 1 2 4 5 9]

Code #2 :

Python3

# Python program explaining
# partition() function
 
import numpy as geek

# input array
in_arr = geek.array([ 2, 0,  1, 5, 4, 9, 3])
print ("Input array : ", in_arr) 

out_arr = geek.partition(in_arr, (0, 3))
print ("Output partitioned array : ", out_arr)

Output:

Input array :  [2 0 1 5 4 9 3]
Output partitioned array :  [0 1 2 3 4 9 5]

numpy.partition() in Python

jana_sayantan

Improve

Article Tags :

Practice Tags :

python

Similar Reads

numpy.packbits() in Python

numpy.packbits() is another function for doing binary operations in numpy.It is used to packs the elements of a binary-valued array into bits in a uint8 array.The result is padded to full bytes by inserting zero bits at the end. Syntax : numpy.packbits(arr, axis=None) Parameters : arr : [array_like]

Numpy recarray.partition() function | Python

In numpy, arrays may have a data-types containing fields, analogous to columns in a spreadsheet. An example is [(a, int), (b, float)], where each entry in the array is a pair of (int, float). Normally, these attributes are accessed using dictionary lookups such as arr['a'] and arr['b']. Record array

SymPy | Partition.next_lex() in Python

Partition.next_lex() : next_lex() is a sympy Python library function that returns the next integer partition, n in lexicographical order. This ordering wraps around [n] if the partition is [1, â€¦, 1]. Syntax : sympy.combinatorics.partitions.Partition.next_lex() Return : next integer partition, n in l

SymPy | Partition.as_dict() in Python

Partition.as_dict() : as_dict() is a sympy Python library function that returns the partition as a dictionary. In this dictionary, the keys are the partition integers. The values are the multiplicity of that integer. Syntax : sympy.combinatorics.partitions.Partition.as_dict() Return : Partition as a

SymPy | Partition.prev_lex() in Python

Partition.prev_lex() : prev_lex() is a sympy Python library function that returns the previous integer partition, n in lexicographical order. This ordering wraps around [n] if the partition is [1, â€¦, 1].Â Syntax : sympy.combinatorics.partitions.Partition.prev_lex()Return : previous integer partition

SymPy | Partition.RGS_enum() in Python

Partition.RGS_enum() : RGS_enum() is a sympy Python library function that calculates the total number of restricted growth strings (string where a[i] is the block in which element i occurs) which is possible for a superset of size n. Restricted growth strings - string in which each character, ai res

Python | sympy.partition() method

With the help of sympy.partition() method, we can find Partition number in SymPy. partition(n) - The Partition numbers are a sequence of integers pn that represent the number of distinct ways of representing n as a sum of natural numbers (with order irrelevant). The generating function for pn is giv

Python | sympy Partition() method

With the help of sympy.combinatorics.partitions.Partition() method, we are going to separate the members of arrays into one tuple by using sympy.combinatorics.partitions.Partition() method. Syntax : sympy.combinatorics.partitions.Partition(array) Return : Return a members of arrays into one tuple. E

SymPy | Partition.conjugate() in Python

Partition.conjugate() : conjugate() is a sympy Python library function that returns the conjugate partition of the argumented partition. Syntax : sympy.combinatorics.partitions.Partition.conjugate() Return : conjugate partition Code #1 : conjugate() ExamplePython3 1=1 # Python code explaining # SymP

NumPy Array in Python

NumPy (Numerical Python) is a powerful library for numerical computations in Python. It is commonly referred to multidimensional container that holds the same data type. It is the core data structure of the NumPy library and is optimized for numerical and scientific computation in Python. Table of C