Python:具有可变长度的参数列表的函数

时间:2013-04-18 15:34:18

标签: python matplotlib

我正在尝试绘制(在同一图上)来自Excel工作表的一些数据,并且我想要一个可变长度的字符串列表作为输入,对应于不同的材料。我收到以下错误:  TypeError:'NoneType'对象不可迭代 我真的不明白为什么。这是代码:

import xlrd
import matplotlib.pyplot  as plt
from numpy import *
def transmittance(*glass):
    wb=xlrd.open_workbook('schott_optical_glass_catalogue_excel_december_2012.xls')
    sheet1=wb.sheet_by_index(0)
    transm_index=[]  #lista vuota
    plt.figure(1)
    plt.xlabel('wavelength $\lambda$[nm]')
    plt.ylabel('transmittance')
    for item in glass:
        for i in range(sheet1.nrows):
            if sheet1.cell_value(i,0)==glass:
                reversed_transmission=sheet1.row_values(i,37,67)
                transm_index=reversed_transmission[::-1]
                new_transm_index=[float(ni)  for ni in transm_index ]
    wavel_range=sheet1.row_values(3,37,67)
    temp_wavel= [k.split('/')[1] for k in wavel_range]
    wRange=map(int,temp_wavel[::-1])
    plt.plot(wRange,new_transm_index, 'r-')
    plt.grid(True, which="both")
    plt.show()
    return new_transm_index, wRange

if __name__=='__main__':
    new_transm_index=transmittance('N-BASF64','N-BK7')
    print 'get tuple length and glass name: ' ,new_transm_index

1 个答案:

答案 0 :(得分:0)

我无法重现您在问题中描述的TypeError(如果我在没有任何参数的情况下调用transmittance(),我可能会得到)。但是,当我使用我通过Google找到的同名XLS文件调用您的函数时,我会遇到两个不同的错误。

  • 您迭代glass中的项目,但然后与整个列表而不是当前item进行比较
  • 当您创建new_transm_index列表时,您不能只转换为float,因为表中有一些空字符串;在这种情况下,我会假设该值为零。

最后,如果您希望new_transm_indexglass中的每个项目保留一个列表(如评论中所述),您应该使用字典,将项目(键)映射到相应的列表(值)。

...
new_transm_index = {} # create dictionary
for item in glass:
    for i in range(sheet1.nrows):
        if sheet1.cell_value(i, 0) == item: # compare to item, not to list
            ...
            # do not cast ' ' to float, but use 0 if not of type float
            new_transm_index[item] = [ni if type(ni) == float else 0 for ni in transm_index]
...
for item in glass: # add plots for all items to common diagram
    plt.plot(wRange,new_transm_index[item])
plt.grid(True, which="both")
plt.show()
...