本文整理汇总了Python中numpy.sort_complex方法的典型用法代码示例。如果您正苦于以下问题：Python numpy.sort_complex方法的具体用法？Python numpy.sort_complex怎么用？Python numpy.sort_complex使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类numpy的用法示例。
在下文中一共展示了numpy.sort_complex方法的5个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: sort_complex
# 需要导入模块: import numpy [as 别名]
# 或者: from numpy import sort_complex [as 别名]
def sort_complex(a):
"""Sort a complex array using the real part first,
then the imaginary part.
Args:
a (cupy.ndarray): Array to be sorted.
Returns:
cupy.ndarray: sorted complex array.
.. seealso:: :func:`numpy.sort_complex`
"""
if a.dtype.char in 'bhBHF':
a = a.astype('F')
else:
a = a.astype('D')
a.sort()
return a 示例2: test_sort_real
# 需要导入模块: import numpy [as 别名]
# 或者: from numpy import sort_complex [as 别名]
def test_sort_real(self, type_in, type_out):
# sort_complex() type casting for real input types
a = np.array([5, 3, 6, 2, 1], dtype=type_in)
actual = np.sort_complex(a)
expected = np.sort(a).astype(type_out)
assert_equal(actual, expected)
assert_equal(actual.dtype, expected.dtype) 示例3: test_sort_complex
# 需要导入模块: import numpy [as 别名]
# 或者: from numpy import sort_complex [as 别名]
def test_sort_complex(self):
# sort_complex() handling of complex input
a = np.array([2 + 3j, 1 - 2j, 1 - 3j, 2 + 1j], dtype='D')
expected = np.array([1 - 3j, 1 - 2j, 2 + 1j, 2 + 3j], dtype='D')
actual = np.sort_complex(a)
assert_equal(actual, expected)
assert_equal(actual.dtype, expected.dtype) 示例4: sort_complex
# 需要导入模块: import numpy [as 别名]
# 或者: from numpy import sort_complex [as 别名]
def sort_complex(a):
"""
Sort a complex array using the real part first, then the imaginary part.
Parameters
----------
a : array_like
Input array
Returns
-------
out : complex ndarray
Always returns a sorted complex array.
Examples
--------
>>> np.sort_complex([5, 3, 6, 2, 1])
array([ 1.+0.j, 2.+0.j, 3.+0.j, 5.+0.j, 6.+0.j])
>>> np.sort_complex([1 + 2j, 2 - 1j, 3 - 2j, 3 - 3j, 3 + 5j])
array([ 1.+2.j, 2.-1.j, 3.-3.j, 3.-2.j, 3.+5.j])
"""
b = array(a, copy=True)
b.sort()
if not issubclass(b.dtype.type, _nx.complexfloating):
if b.dtype.char in 'bhBH':
return b.astype('F')
elif b.dtype.char == 'g':
return b.astype('G')
else:
return b.astype('D')
else:
return b 示例5: test_sort_complex
# 需要导入模块: import numpy [as 别名]
# 或者: from numpy import sort_complex [as 别名]
def test_sort_complex(self):
self.check(np.sort_complex)