振幅・位相のリストから複素数のリストを作成する

Question

#質問内容
エクセル上で準備した振幅　位相データをpython上でfor文と関数を用いて複素数変換を行いたいです。
#Excelデータ（一部）
0.487436891 0
0.653218389 134.6398315
0.193872914 166.9507599
0.202087417 123.1289139
0.583534479 154.6550446
0.128547326 56.14105988
0.12943536 -154.5452881
0.3214885 177.0462799
0.029101485 138.5561066
0.192596287 -175.8987427
...
#コード

def re(a,b):
    c=a*math.cos(b*math.pi/180)
    return c
def im(x,y):
    z=x*math.sin(y*math.pi/180)
    return z
mag=[]
ph=[]
v=[]

with open(Mypath,encoding=("utf-8-sig")) as f:
         reader=csv.reader(f, quoting=csv.QUOTE_NONNUMERIC)
         print(reader)
         print(type(reader))
         for row in csv.reader(f):
             mag.append(float(row[0]))
             ph.append(float(row[1]))
 　　　　　　　v.append[row]=complex(re(mag[row],ph[row]),im(mag[row],ph[row]))

#出力結果
v.append[row]=complex(re(mag[row],ph[row]),im(mag[row],ph[row]))

TypeError: list indices must be integers or slices, not list

#試したこと
v[row]=complex(re(mag[row],ph[row]),im(mag[row],ph[row]))
このようにappendを使わなくても同様のエラーが出てくるのですが、エラーの意味が解らず困っています。

#補足
振幅データはmagに、位相データはphに格納し、そのデータ数は800個です。
これらのデータを関数を用いて1次元のりすとに複素数で格納したいです。

Comments

[ppaul]

エラーの原因はエラーメッセージにあるように、リストのインデックスにリストをつかっていることです。
どうすれば解決するかをしりたければ、csvファイルの最初の5行ぐらいを載せてください。
それがないと回答は難しいです。

Answer 1

最小限の変更で済ますなら、

with open(Mypath,encoding=("utf-8-sig")) as f:
         reader=csv.reader(f, quoting=csv.QUOTE_NONNUMERIC)
         print(reader)
         print(type(reader))
         for row in csv.reader(f):
             a_magnitude = float(row[0])
             mag.append(a_magnitude)
             a_phase = float(row[1])
             ph.append(a_phase)
             v.append(complex(re(a_magnitude,a_phase),im(a_magnitude,a_phase)))

です。

ただし、このコードではエクセルの.xlsxファイルは読めません。
入力ファイルをエクセルで開き、csvとして保存する、で保存してできる.csvファイルを指定してください。

エクセルの.xlsxファイルを読みたければopenpyxlを使って読むか、pandasを使って読むことになるので大幅な変更が必要です。

pandas版を作ってみました。原型は留めていないですね。

import pandas as pd
import cmath

df = pd.read_excel('complex.xlsx', header=None)
df.columns = ['mag', 'ph']
df['v'] = df.apply(lambda row: cmath.rect(row['mag'], row['ph']*cmath.pi), axis=1)
print(df)
mag = list(df['mag'])
ph = list(df['ph'])
v = list(df['v'])
print(mag)
print(ph)
print(v)

実行結果は以下です。

>>> import pandas as pd
>>> import cmath
>>>
>>> df = pd.read_excel('complex.xlsx', header=None)
>>> df.columns = ['mag', 'ph']
>>> df['v'] = df.apply(lambda row: cmath.rect(row['mag'], row['ph']*cmath.pi), axis=1)
>>> print(df)
        mag          ph                   v
0  0.487437    0.000000  0.487437+0.000000j
1  0.653218  134.639831 -0.277814+0.591197j
2  0.193873  166.950760 -0.191558+0.029871j
3  0.202087  123.128914 -0.185739-0.079625j
4  0.583534  154.655045 -0.273125+0.515669j
5  0.128547   56.141060  0.116130+0.055120j
6  0.129435 -154.545288 -0.018354-0.128128j
7  0.321489  177.046280 -0.318097-0.046578j
8  0.029101  138.556107 -0.005103+0.028651j
9  0.192596 -175.898743  0.182933+0.060239j
>>> mag = list(df['mag'])
>>> ph = list(df['ph'])
>>> v = list(df['v'])
>>> print(mag)
[0.487436891, 0.653218389, 0.193872913999999, 0.202087417, 0.583534479, 0.128547326, 0.12943536, 0.3214885, 0.029101485, 0.192596287]
>>> print(ph)
[0.0, 134.639831499999, 166.9507599, 123.1289139, 154.6550446, 56.14105988, -154.5452881, 177.0462799, 138.5561066, -175.898742699999]
>>> print(v)
[(0.487436891+0j), (-0.2778139468650384+0.5911968154980412j), (-0.19155787387867684+0.02987118574700291j), (-0.18573943062119574-0.07962529762736151j), (-0.27312543609590767+0.5156694525946071j), (0.11613020926096146+0.05511977429884463j), (-0.018353576473760323-0.12812750933718892j), (-0.31809650787860944-0.04657754080760128j), (-0.0051030261112949655+0.028650576847816987j), (0.18293344364875225+0.06023856705629006j)]

Answer 2

最後の行の意味が突然おかしいですね。

    for row in csv.reader(f):
        mag.append(float(row[0]))   # 強度
        ph.append(float(row[1]))    # 位相
        v.append[row]=complex(re(mag[row],ph[row]),im(mag[row],ph[row]))

rowには「index番号」が入っているわけではないので、「[row]」のように書くのは誤りです。
また、直前で入れたmagとphから値を取り出して使うという意図だと思いますが、あまり筋がよくないので、一旦変数で受けてそれぞれに入れることにします。変数で受けるのがいやであれば、そのままfloat(row[0]))を使ってもいいと思います。

    for row in csv.reader(f):
        magnitude = float(row[0])
        phase = float(row[1]) 
        mag.append(magnitude)   # 強度
        ph.append(phase)    # 位相
        v.append(complex(re(magnitude,phase),im(magnitude,phase)))