Python：列表或浮點數組或浮點數

Question

我有一個函數需要輸入作爲類型float進一步計算。 不幸的是，輸入類型範圍：

1. list
2. float
3. numpy.array

我不滿意我的解決辦法，我不知道是否有更好的解決方案。

def get_correct_float(x): 
    try: 
     if len(x) == 1: 
      return x[0] 
    except: 
     return x  

a = [.5] 
b = .5 
c = np.array([.5]) 

get_correct_float(a) == get_correct_float(b) == get_correct_float(c) 
TRUE 
get_correct_float(a) 
0.5 

Answer 1

使用isinstance，

>>> import numpy as np 
>>> l = [[.5], .5, np.array([.5])] 
>>> f = [x if isinstance(x, float) else x[0] for x in l] 
>>> f 
[0.5, 0.5, 0.5] 

---

def get_correct_float(x): 
    return x if isinstance(x, float) else x[0] 

---

以前的答案，

import numpy as np 

def get_correct_float(x): 
    if isinstance(x, list): 
     return x[0] 
    elif isinstance(x, np.ndarray): 
     return x[0] 
    else: 
     return x