在for循環中使用python numpy linspace

Question

我正在嘗試在for循環中使用linspace。 我希望0到10之間的間隔為0.5。看來z_bin執行正確。

我的問題：如何在for循環中正確使用linspace代替注釋行旁邊顯示的range函數？ 從整數處理轉向小數處理時，循環中需要更改什么？

z_bin = numpy.linspace (0.0,10.0,num=21) 
print 'z_bin: ', z_bin
num = len(z_bin)

grb_binned_data = []
for i in numpy.linspace(len(z_bin)-1): #range(len(z_bin)-1):
    zmin = z_bin[i]
    zmax = z_bin[i+1]
    grb_z_bin_data = []
    for grb_row in grb_data:
        grb_name = grb_row[0]
        ra = grb_row[1]
        dec = grb_row[2]
        z = float(grb_row[3])
        if z > zmin and z <= zmax:
            grb_z_bin_data.append(grb_row)
    grb_binned_data.append(grb_z_bin_data) 

Answer 1

您不應該使用linspace的輸出來索引數組！ linspace生成浮點數，並且數組索引需要一個整數。

從您的代碼看來，您真正想要的是這樣的：

z_bin = numpy.linspace(0.0, 10.0, 21)

for i in range(len(z_bin)-1):
    zmin = z_bin[i]
    zmax = z_bin[i+1]

    # do some things with zmin/zmax

zmin和zmax仍然會浮點數（十進制）。

Answer 2

一個更簡單的方法是使用zip函數：

z_bin = numpy.linspace(0.0, 10.0, 21)

for z_min, z_max in zip(z_bin[:-1], z_bin[1:]):
    print z_min, z_max