質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.91%

numpyの1次配列の複素数の複雑な式を「実数+複素数」に簡略化するには?

解決済

回答 1

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 192

Junpei-O

score 12

以下のような、複雑な複素数式を「実数+複素数」という形に計算した形にしたいです。
どのように行えば良いでしょうか。ご教示いただけましたら幸いです。
よろしくお願い致します。

#a0~a6は800行1列の2次元ndarrayです。要素の中には複数数値をとるものもあります。refが最終結果です。
fai=(a0+1j*a1)*2*sp.pi/a0*d*np.cos(a3+1j*a4)
ref=(a5+a6*sp.exp(2j*fai))/(1+a5*a6*sp.exp(2j*fai))

#1行分のa0~a6の例、上から順にa0,a1,,,,a6と示しております。
#[ 7.00000000e-07+0.00000000e+00j  1.01015674e+00+0.00000000e+00j
#  1.24434645e-05+0.00000000e+00j  1.57077824e+00+0.00000000e+00j
# -8.31325978e-01+0.00000000e+00j  9.99997866e-01+0.00000000e+00j
# -1.00000000e+00+8.27180613e-25j]

行ったこと

  • 既にsympy.powsimpやsympy.radsimp、sympy.simplifyは試しています。しかし、望んだようにはうまくいきません。

  • .realや.imagを使って実数部、虚数部を抜き出す方法
    一つの要素の結果ですが、以下のようになってしまい、望んだ結果になりませんでした。他の要素も同様複雑な結果となりました。

# print(ref.real)の抜粋
[(0.999999367376849 + (-1.0 + 2.06795153138257e-25*I)*exp(0.002002002002002*I*pi*(1.998e-6 + 2.01329802790684*I)*(7.43928380851181e-6 + 0.937418356333597*I)))/(1 + (-0.999999367376849 + 2.06795022314856e-25*I)*exp(0.002002002002002*I*pi*(1.998e-6 + 2.01329802790684*I)*(7.43928380851181e-6 + 0.937418356333597*I)))]
  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

checkベストアンサー

0

型が複素数の numpy 配列は ndarray.real で実部、ndarray.imag で虚部だけ取得できます。

import numpy as np

x = np.array([1 + 1j, 2 + 2j, 3 + 4j])
print(x)  # [1.+1.j 2.+2.j 3.+4.j]
print(x.dtype)  # complex128

print(x.real)  # [1. 2. 3.]
print(x.imag)  # [1. 2. 4.]

追記

import numpy as np
import sympy as sy

x = sy.Symbol("x")
sol = sy.solve(2 * x ** 2 + 4 * x + 5, x)
print(sol)

# 数値に変換してから分解
sol_num = np.fromiter(map(sy.N, sol), dtype=complex)
print(sol_num)
print(sol_num.real, sol_num.imag)

# 記号のまま分解
print([x.as_real_imag() for x in sol])

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/07/21 18:14 編集

    追記しました。
    sympy の式なら、sympy.N で numpy 配列に変換してから、real、imag で取得するか、as_real_imag() で記号のまま分解することもできます。

    キャンセル

  • 2020/07/21 18:22 編集

    ご対応ありがとうございます。以下のようにすることで自己解決できました。
    tiitoi様の最初の回答で、型がcomplexの場合というところが気になり、調べたところrefがpythonオブジェクトになっておりました。astypeでcomplexにして、.real、.imagをやるとうまく行きました。
    numyの要素にsp.expを使っています。sp.expを使っているのは、np.expでエラーが出たためです。まだ検算を行っていませんが、問題になるのでしょうか。

    キャンセル

  • 2020/07/21 18:37

    > まだ検算を行っていませんが、問題になるのでしょうか。

    おそらく問題ないとは思いますが、念の為、検算してみるといいかと思います。

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.91%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る