3

以下にリストされている Matlab コマンドを Python で実装したいと考えています。Python で Matlab の同等のコマンドを理解することはできますが、正確な結果が得られません。誰かがそうするのを手伝ってくれませんか。

MATLAB コード:

n0 = 3
n1 = 1
n2 = 5
n = [n1:n2]
>> 1 2 3 4 5
x = [(n - n0) == 0]
>> 0 0 1 0 0

パイソンコード:

import numpy 
n0 = 3
n1 = 1
n2 = 5
n = r_[n1:n2+1]
>> [1 2 3 4 5]
x = r_[(n-n0) == 0]
>> [False False True False False]

したがって、x はブール データ型 "[array([False, False, True, False False], dtype=bool)]" を持つ私の配列です。結果がMatlabとまったく同じになるように、0または1の形式で結果を返す最後のコマンドを作成するにはどうすればよいですか。

4

1 に答える 1

2

リスト内包表記を使用して bool を int に変換します。

[int(val) for val in x]
于 2012-08-10T09:26:04.393 に答える