python - 使用 SPSS Python 脚本获取列的值

Question

我有一个 SPSS Python 脚本，它遍历一个表并读取特定列的每一行值，并相应地设置参数值。目前，我正在使用 getValueAt(rowIndex, colIndex) 访问这些值。但是，必须引用列索引而不是列名并不理想，因为表列可能会移动。有没有一种方法可以根据列名引用值？

示例代码

diagram = modeler.script.diagram()

for i in range(nrows):

    jobid = job_rowset.getValueAt(i, 0);
    city = job_rowset.getValueAt(i, 3);
    country = job_rowset.getValueAt(i, 4);

    diagram.setParameterValue ('BU', bu)
    diagram.setParameterValue ('City_Param', city)
    diagram.setParameterValue ('CountryID_Param', country)

感谢任何帮助!谢谢!

Answer 1

假设第一行将包含名称(或者您有一些可用的东西来指定列顺序)，您可以构建一个字典，其中列名作为键，列号作为值。

这会给出类似的东西:

name_key_dict = dict()
for i in range(ncols):
    name_key_dict[colnames[i]] = i # Assuming that names is the ordered list of columns
# I would add a check here that you have the required columns

param_col_list = [ # This constructs a list of parameters vs column numbers
    ('BU', name_key_dict.get('Bu_Col')),
    ('City_Param', name_key_dict.get('City')), 
    ('CountryID_Param', name_key_dict.get('Country Code')), 
]
for row in job_rowset: # Assuming job_rowset is an iterable with a member of getValue
    for (param, col) in param_col_list:
        diagram.setParameterValue(param, row.getValue(col))