1. 程式人生 > >Python浮點數range

Python浮點數range

 問題提出:最近在用matplotlib定義座標軸,需要會出現型如包含[0.1,0.2...]這樣的等差序列,實現這個函式略有難度,以下是我的解決過程,和解決辦法,希望對大家有所幫助。
 最初覺得這個問題很簡單,  雖然python的range是不支援浮點數的,np可以實現出現下面的程式碼```

import numpy as np
print(list(map(lambda a:str(a),np.arange(0,0.8,0.1))))
#out:[‘0.0’, ‘0.1’, ‘0.2’, ‘0.30000000000000004’, ‘0.4’, ‘0.5’, ‘0.6000000000000001’, ‘0.7000000000000001’]

顯然不滿足要求,要求更高精度。想過使用 decimal所以嘗試這段程式碼

from decimal import Decimal
def Demyrange(i:float,j:float,k:float)->list:
    lista=[]
    while i<j:
        lista.append(str(float(i)))
        i=Decimal(i)
        k=Decimal(k)
        i+=k
    return lista
print(Demyrange(0,0.8,0.1)) 
#out:['0.0', '0.1', '0.2', '0.3', '0.4', '0.5', '0.6', '0.7000000000000001', #'0.8']

還是有點問題。這樣,自然想到format於是出現這段程式碼:

def myrange(i:float,j:float,k=1)->list: 
    lista=[]
    while i<j:
        lista.append(format(i, "0.1f"))
        i+=k
    return lista
print(myrange(0,0.8,0.1)) 
#out: ['0.0', '0.1', '0.2', '0.3', '0.4', '0.5', '0.6', '0.7', '0.8']

好了,基本解決問題,但是步長每次需要判斷是很討厭的事情,所以修改如下

def myrange(i:float,j:float,k=1)->list:
    xlen=str((len(str(k-int(k)))-2)/10)+"f" #根據k步長,判斷format位數公式小數點位數/10+"f"
    print("xlen=",xlen)
    lista=[]
    while i<j:
        lista.append(format(i, xlen))
        i+=k
    return lista

print(myrange(0,0.8,0.2))# out: xlen= 0.1f ['0.0', '0.2', '0.4', '0.6']
print(myrange(0,0.08,0.02)) # xlen= 0.2f  ['0.00', '0.02', '0.04', '0.06']
print(myrange(0,0.008,0.002))# ['0.000', '0.002', '0.004', '0.006']
print(myrange(0,0.0008,0.0002))# ['0.0000', '0.0002', '0.0004', '0.0006']
這樣基本滿足要求:xlen=str((len(str(k-int(k)))-2)/10)+"f"的意思是:
根據步長自動設定格式