將matlab程序轉換爲等效的python代碼

Question

想在Python中實現下面列出的Matlab命令。我能夠找出在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 

Python代碼：

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是我的數組布爾數據類型「陣列（[FALSE，FALSE，TRUE，假假]，D型=布爾）]」。我怎樣才能使我的最後一個命令返回結果的形式爲0或1的結果是完全一樣的Matlab。

Answer 1

使用列表解析轉換布爾爲int：

[int(val) for val in x]