關於代碼的條件

Question

我有一個數組的數據集。我想取出那些數值位於（0.5,0.6）的數組。

我這樣做是對前面只有不到0.4

的代碼是

c_subset = np.array(c)[np.where(np.array(c)<0.4)]

誰能幫助到如何修改上述行的間隔條件？

Answer 1

使用numpy logical and：

>>> c = [0.50, 0.52, 0.54, 0.56, 0.58, 0.60] 
>>> a = np.array(c) 
>>> a[(0.5 < a) & (a < 0.6)] 
array([0.52, 0.54, 0.56, 0.58]) 

注意numpy.where不必要。

Answer 2

如果您的原始數據不使用numpy和一個簡單的列表輸出足夠，你可以這樣做：

c_subset = [x for x in c if 0.5 < x < 0.6]