Question

如何在Python中创建一组复数？

在C ++ STL中，我们可以编写以下代码：

class Complex {

      public:
      int re, 
          im;

      float getModule() {

            return sqrt(re * re + im * im);
      } 
};

vector< Complex > vec;

但是在Python中？

Answer 1

非常简单：

Python有本机列表和本机复杂类型，所以：

c = complex(real,imag)

或只是

c = 1 + 2j

可以创建一个;

complexes = [ complex(i, i) for i in range(100) ]

在列表复合体中创建了数千个复杂值。

你可能想看看numpy：

import numpy
arr = numpy.ndarray(1000, dtype=numpy.complex128)

Answer 2

Python内置了对复数的支持。你可以这样输入：

>>> a = 2 + 3j # or: complex(2,3)
>>> a
(2+3j)
>>> type(a)
<type 'complex'>
>>> a.real
2.0
>>> a.imag
3.0
>>>

对于容器，在Python中，您可以从list开始：

>>> complex_nums_list = [2+3j, 3+4j, 4+5j]
>>> complex_nums_list
[(2+3j), (3+4j), (4+5j)]

或者您可以使用numpy.array，这更适合数字应用。

Answer 3

您只需创建值列表：

vec = [1+2j, 3+4j]

Answer 4

您可以使用内置的complex类。

或者只使用复杂的文字：Python使用j作为虚构单位。

z = complex(3, 4)
print(z, z.real, z.imag)

z = 3 + 4j
print(z)

<强>输出

(3+4j) 3.0 4.0
(3+4j)

复杂构造函数也接受关键字参数，因此您可以执行

z = complex(real=3, imag=4)

以任意顺序使用args。这也意味着如果你愿意，你甚至可以在dict中传递args：

d = {'real': 3, 'imag': 4}
z = complex(**d)

还有一个用于复杂参数的数学函数的内置cmath模块。

Answer 5

实际上，我想根据他们的模块对数字进行排序。轮到我了。

import math

class Complex:
  def __init__( self, a, b ): 
      self.a = a
      self.b = b

  def getModule( self ):
      return math.sqrt( self.a**2 + self.b**2 )

  def __str__( self ):
  ''' Returns complex number as a string '''
      return '(%s + i %s)' % (self.a, self.b)

  def add(self, x, y):
      return Complex(self.a + x, self.b + y) 

  def sub(self, x, y):
      return Complex(self.a - x, self.b - y) 


  #input = [[2, 7],[5, 4],[9, 2],[9, 3],[7, 8], [2, 2], [1, 1]]

  # Read the input from a given file complex.in, 
  # in fact is a matrix with Nx2 dimensions
  # first  line re1 im1 
  # second line re2 im2
  # ...
  #Example complex.in
  #5
  #2 7
  #5 4
  #9 2
  #9 3
  #7 8

  f = open('complex.in','r')

  input = [map(int, line.split(' ')) for line in f]

  del input[0]

  num = len( input )

  complexes = [ Complex( i[ 0 ], i[ 1 ] ) for i in input ]

  def swapp(c, a, b):

      c[ a ], c[ b ] = c[ b ], c[ a ] 

  def sort( c ):

      swapped = 1

      for i in range(num - 1, 0, -1):

      swapped = 1

      for j in range(0, i):

          if c[ j ].getModule() > c[ j + 1 ].getModule():

             swapped = 0

             swapp(c, j, j + 1)

      if swapped:

         break   

  sort( complexes )

  f = open('complex.out','w')

  for c in complexes:

      f.write('%s\n' % c)     
      print c

Python中的复数数组

5 个答案: