编写引用特定列的函数

def get_column_values(data, column):
    # data为数据集或表格，column为要引用的列的索引或名称
    column_values = []
    if isinstance(column, int):
        # 通过索引引用列
        for row in data:
            column_values.append(row[column])
    elif isinstance(column, str):
        # 通过名称引用列
        column_index = data[0].index(column)
        for row in data[1:]:
            column_values.append(row[column_index])
    return column_values

data = [
    ['Name', 'Age', 'Gender'],
    ['John', 25, 'Male'],
    ['Emily', 30, 'Female'],
    ['Michael', 35, 'Male']
]

# 引用第二列（索引为1）的值
column_values = get_column_values(data, 1)
print(column_values)
# 输出：[25, 30, 35]

# 引用名为'Gender'的列的值
column_values = get_column_values(data, 'Gender')
print(column_values)
# 输出：['Male', 'Female', 'Male']