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

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

新規登録して質問してみよう
ただいま回答率
85.50%
NumPy

NumPyはPythonのプログラミング言語の科学的と数学的なコンピューティングに関する拡張モジュールです。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

2回答

1081閲覧

ndarrayのufuncの引数にリストを渡して利用したい

yoh_chan

総合スコア13

NumPy

NumPyはPythonのプログラミング言語の科学的と数学的なコンピューティングに関する拡張モジュールです。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2017/09/18 05:17

編集2017/09/19 10:23

###前提・実現したいこと
Pythonで、Numpyのfrompyfuncを利用してifを含む処理を配列に適用したいと考えています。
その処理の中で、引数の一つであるリストの要素を参照したいのですが、ufuncにすることによって、ブロードキャストの際にエラーが出ます。

以下の例ですと、引数のリストをそれぞれの要素に持つ(3,4)型の配列になって欲しいのですが、(当然ながら)リスト自体をブロードキャストしようとして失敗しているという状況です。

処理が遅いと聞くのでfor文で回すのは避けたいと思っているのですが、やはりfor文で処理するしかないでしょうか。お知恵をお借りしたいです。


2017-09-19追記
実際のリスト(または辞書)はデータベースから取得したパラメータの一覧であり、数十~数百程度の要素をもつテーブルで、配列の要素の値を受けて異なるインデックスを参照する必要があります。テーブルのデータなのでリストではなくndarrayであっても構わないのですが、いずれにしてもブロードキャストで失敗します。


###該当のソースコード

Python

1import numpy as np 2 3 4# ufunc化したい関数の定義 5def hoge(a, list_b): 6 if a % 2 == 0: # 偶数 7 x = a * list_b[0] 8 else: # 奇数 9 x = a * list_b[1] 10 return x 11 12 13# ufunc化 14uf_hoge = np.frompyfunc(hoge, 2, 1, dtype=int) 15 16# 変数の定義 17a = np.arange(12).reshape(3,4) 18list_b = [10, -10] 19 20# 処理 21res = uf_hoge(a,list_b) 22 23

###発生している問題・エラーメッセージ

ValueError Traceback (most recent call last) <ipython-input-14-c2794e861f7f> in <module>() ----> 1 uf_hoge(a,list_b) ValueError: operands could not be broadcast together with shapes (3,4) (2,)

###補足情報(言語/FW/ツール等のバージョンなど)
Python 2.7.5
Numpy 1.12.1

気になる質問をクリップする

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

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

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答2

0

ベストアンサー

引数にリストを渡すのをやめて、スカラー値を2つ渡すようにするという方法では駄目でしょうか?

Python

1import numpy as np 2 3def hoge(a, b0, b1): 4 if a % 2 == 0: 5 x = a * b0 6 else: 7 x = a * b1 8 return x 9 10a = np.arange(12).reshape(3,4) 11uf_hoge = np.frompyfunc(hoge, 3, 1, dtype=int) 12res = uf_hoge(a,*list_b)

あと、この程度の関数であれば、numpy.frompyfunc()ではなくnumpy.where()で書き換えた方が簡単な気がします。

Python

1import numpy as np 2 3a = np.arange(12).reshape(3,4) 4list_b = [10, -10] 5res = np.where(a % 2 == 0, a * list_b[0] ,a * list_b[1])

【補足】
lambdaで引数をバインドした例

Python

1import numpy as np 2 3def hoge(a, list_b): 4 if a % 2 == 0: # 偶数 5 x = a * list_b[0] 6 else: # 奇数 7 x = a * list_b[1] 8 return x 9 10 11# 変数の定義 12a = np.arange(12).reshape(3,4) 13param_b = [10, -10] 14param_c = [100, -100] 15param_d = [1000, -1000] 16 17# param_b を引数にした関数をvectorize 18uf_hoge_b = np.vectorize(lambda a: hoge(a, param_b)) 19 20# param_c を引数にした関数をvectorize 21uf_hoge_c = np.vectorize(lambda a: hoge(a, param_c)) 22 23# param_d を引数にした関数をvectorize 24uf_hoge_d = np.vectorize(lambda a: hoge(a, param_d)) 25 26 27# 処理 28print(uf_hoge_b(a)) 29print(uf_hoge_c(a)) 30print(uf_hoge_d(a))

投稿2017/09/19 00:30

編集2017/09/19 10:50
magichan

総合スコア15898

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

yoh_chan

2017/09/19 10:07

ご回答いただき、ありがとうございます。情報が不十分で申し訳ありません。 例示したものは極めて簡略化しているのですが、実際のリスト(または辞書)はデータベースから取得したパラメータの一覧であり、数十~数百程度の要素をもつテーブルで、配列の要素の値を受けて異なるインデックスを参照する必要があります。 一つの関数に対して引数として与えるリスト(これは複数の場合もあります)のサイズはそれぞれ決まっているので、それらを全て分解してスカラー値として与えることも可能ではありますが、与えるパラメータ数と条件分岐が非常に多くなってしまうので避けたいところです。 当然、配列の要素全てに同一のリストを格納するというのは不自然なので、可能ならリストそのものをufunc内で利用できればよいのですが、引数は全て自動的にブロードキャストされてしまうので難儀しています。 配列の値に応じて処理を条件分岐する方法がufunc以外にもあればよいのですが、、、
magichan

2017/09/19 10:50

そういうことなのですね・。失礼しました。 ジャストアイデアなのですが、そのパラメータ群を frompyfunc の第2引数に渡すことを諦めて、一旦 lambda 関数などでバインドした後に ベクトル関数化するというのはいかがでしょうか。(サンプルを追加しました。) パラメータ毎に関数を作らなければならない気持ち悪さは残りますけれど。
yoh_chan

2017/09/20 12:17

lambda関数をそのような形で使うなんて、まったく思いつきもしないことでした…ありがとうございます。 ただ、違う着目点ですが、補足いただいたコードを眺めていてnp.vectorizeも使えるんだったなあ、と思って調べてみたら、こちらにはexcludedというドンピシャなオプションがあることがわかりました。 提案いただいた解法ではありませんが、おかげさまでひとまず望む結果を得ることができました。ありがとうございます。 関数の中には出力が複数になるものもあるので、これをどうするかが次の格闘すべき問題になりそうですが…^^;
guest

0

いただいた示唆をもとに、自己解決しました。

np.vectorize()にはexcludedオプションでブロードキャストを回避できるため、これを利用することで期待する挙動を得ることができました。

Python

1import numpy as np 2 3 4# ufunc化したい関数の定義 5def hoge(a, list_b): 6 if a % 2 == 0: # 偶数 7 x = a * list_b[0] 8 else: # 奇数 9 x = a * list_b[1] 10 return x 11 12 13# vectorize(ブロードキャスト除外パラメータをインデックスで指定) 14uf_hoge = np.vectorize(hoge, excluded=[1]) 15 16# 変数の定義 17a = np.arange(12).reshape(3,4) 18list_b = [10, -10] 19 20# 処理 21res = uf_hoge(a,list_b) 22print(res)

結果

Python

1[[ 0 -10 20 -30] 2 [ 40 -50 60 -70] 3 [ 80 -90 100 -110]]

(ただ、戻り値が複数ある関数の場合は適用できませんが…)

投稿2017/09/20 12:18

yoh_chan

総合スコア13

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

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

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問