if語句，而切片

Question

我有一個數字，我切片列表，使得（使用例如數據）：

midpoint = [[0.2], [0.5], [0.6]] 
Values = [[0.1, 0.3, 0.6, 0.8], [0.2,0.3,0.5,0.7], [0.2,0.5,0.6,0.9]] 
numbers = numpy.arange(0, len(Values), 1) 

我然後有一個循環，發現從所述第一值的numbers索引位置即大於或小於midpoint的值。我有：

for i in range(len(Values)): 
    indexgtr = numbers[Values[i]>=midpoint[i]][-1] # The position of the first number larger than the midpoint 
    indexlt = numbers[Values[i]<=midpoint[i]][0] # The position of the first number larger than the midpoint 

然後，我使用這些索引位置從我有一些其他列表返回值。

我的問題是，有時候沒有價值已超過中點所以indexgtr或大或小或indexlt返回一個空列表[]和我得到的錯誤IndexError: index 0 is out of bounds for axis 0 with size 0。

有什麼我可以補充說，將抓住這一點，並將值0替換爲空列表？

Answer 1

可以得到想要的結果向量化和使用argmax沿着第二軸：

>>> midpoint = np.array([[0.2], [0.5], [0.6], [0.3]]) 
>>> values = np.array([[0.1, 0.3, 0.6, 0.8], 
         [0.2, 0.3, 0.5, 0.7], 
         [0.2, 0.5, 0.6, 0.9], 
         [0.3, 0.1, 0.8, 0.9]]) 

>>> (values > midpoint).argmax(axis=1) # indexgtr vectorized 
array([1, 3, 3, 2]) # first >= occurrence for each row 

>>> (values < midpoint).argmax(axis=1) 
array([0, 0, 0, 1]) # first < occurrence for each row 

---

注：我已經取代了<=和>=與<和>展現更好的矢量結果。請注意，對於<的情況，第一行沒有任何小於0.2的列，但0已返回（因爲它是第一次出現的情況，因爲它是充滿False的行）。

.argmax(axis=1)找到橫跨第2軸的最大值的位置。由於它是一個布爾數組，它返回第一個True事件。

Answer 2

有多種方法，你可以表達測試：

for i in range(len(Values)): 
    indexgtr = numbers[Values[i]>=midpoint[i]] 
    if indexgtr.shape[0]==0: 
     indexgtr = 0 
    else: 
     indexgtr = indexgtr[-1] 
    indexlt = numbers[Values[i]<=midpoint[i]] 
    if indexlt.shape[0]:  # alt expression 
     indexlt = indexlt[0] 
    else: 
     indexlt = 0 
    # indexlt = indexlt[0] if len(indexlt) else 0 

他們都有點羅嗦，但我不認爲他們是昂貴的。你在這裏沒有使用矢量做任何事情 - 除了從numbers中選擇子值。由於numbers已排序，因此您也可以使用min或max做些事情，而不是選擇第一個或最後一個值。

測試表達與喜歡的東西：

In [39]: x=np.arange(0) 
In [40]: x[0] if len(x) else 0 
Out[40]: 0 

我以爲你會在做一些與此環路indexgtr和indexlt上移動到下一個i之前。