处理列表或单个整数作为参数

Question

函数应基于行名（在本例中为第2列）选择表中的行。它应该能够使用单个名称或名称列表作为参数并正确处理它们。

这就是我现在所拥有的，但是理想情况下不会有重复的代码，并且会聪明地使用诸如异常之类的东西来选择正确的方式来处理输入参数：

def select_rows(to_select):
    # For a list
    for row in range(0, table.numRows()):
        if _table.item(row, 1).text() in to_select:
            table.selectRow(row)
    # For a single integer
    for row in range(0, table.numRows()):
        if _table.item(row, 1).text() == to_select:
            table.selectRow(row)

Answer 1

实际上，我同意AndrewHare的回答，只需传递一个包含单个元素的列表即可。
我会这样做：

def select_rows(to_select):
    # For a list
    for row in range(0, table.numRows()):
        if _table.item(row, 1).text() in to_select:
            table.selectRow(row)

并期望该参数将始终是一个列表-即使它只是一个元素的列表。

记得：

要求宽恕比允许容易。

但是，如果您真的必须接受一个非列表，那么在这种情况下将其变成一个列表呢？

def select_rows(to_select):
    if type(to_select) is not list: to_select = [ to_select ]

    for row in range(0, table.numRows()):
        if _table.item(row, 1).text() in to_select:
            table.selectRow(row)

在单项列表上执行“ in”的性能损失可能不会很高：-)但这确实指出了如果“
to_select”列表可能很长，您可能要考虑做的另一件事：考虑强制转换将其设置为一组，以便查找更高效。

def select_rows(to_select):
    if type(to_select) is list: to_select = set( to_select )
    elif type(to_select) is not set: to_select = set( [to_select] )

    for row in range(0, table.numRows()):
        if _table.item(row, 1).text() in to_select:
            table.selectRow(row)