时间戳

首页 > 解决方案 > 如何检查列表（或数组）上的条件，然后在 Python 中根据该条件对元素进行排序

python - 如何检查列表（或数组）上的条件，然后在 Python 中根据该条件对元素进行排序

问题描述

我有一个包含 n 个元素的复数列表，我想检查有多少元素是实数（虚部为零），然后以从实数开始的方式对我的列表进行排序。

例如下面的列表：

a = [ 7 + 0j, -2 + 3j, -2 - 3j, 5 + 6j, 5 - 6j, -1+ 0j, -8 + 4j, -8 - 4j]

两个元素是实数（第一个元素和第六个元素）我想知道我的列表中有 2 个实数元素，然后我想进行如下排序，以实数开头，其他元素保持不变：

b = [ 7 + 0j, -1+ 0j, -2 + 3j, -2 - 3j, 5 + 6j, 5 - 6j, -8 + 4j, -8 - 4j]

我怎样才能做到这一点？谢谢

标签： pythonarrayslistsortingcomplex-numbers

解决方案

Python 有一个sorted带key参数的函数。

key 参数的值应该是一个函数，它接受一个参数并返回一个用于排序目的的键。

您可以根据检查是否存在非零虚部的 lambda 函数进行排序。

a = [ 7 + 0j, -2 + 3j, -2 - 3j, 5 + 6j, 5 - 6j, -1+ 0j, -8 + 4j, -8 - 4j]
sorted(a, key = lambda x: x.imag != 0) 
#Output: [(7+0j), (-1+0j), (-2+3j), (-2-3j), (5+6j), (5-6j), (-8+4j), (-8-4j)]

推荐阅读