我有一个包含n个元素的复数列表,我想检查有多少个元素是实的(虚部为零),然后以实数开始对我的列表进行排序。
例如,以下清单:
a = [ 7 + 0j, -2 + 3j, -2 - 3j, 5 + 6j, 5 - 6j, -1+ 0j, -8 + 4j, -8 - 4j]
有两个元素是实的(第一个元素和第六个元素),我想知道我的列表中有两个实元素,然后我希望下面的类以实数开头,其他元素保持不变:
b = [ 7 + 0j, -1+ 0j, -2 + 3j, -2 - 3j, 5 + 6j, 5 - 6j, -8 + 4j, -8 - 4j]
我怎么能这么做?谢谢
发布于 2019-01-30 18:00:59
你可以通过比较它们的虚部和零来对数字进行排序。0的排在第一位,剩下的数字按顺序不变。
a = sorted(a, key=lambda x:x.imag!=0)
print(a)
输出
[(7+0j), (-1+0j), (-2+3j), (-2-3j), (5+6j), (5-6j), (-8+4j), (-8-4j)]
若要获取此类条目的数目,请使用
sum(x.imag == 0 for x in a)
发布于 2019-01-30 18:02:45
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)]
https://stackoverflow.com/questions/54446685
复制相似问题