Python如何实现PHP的array_column函数功能？

如何用 python 实现类似 php array_column 方法？

对于需要从嵌套列表中提取特定列值的情况，php 中的 array_column 方法是一个有用的工具。以下是如何用 python 实现类似的效果：

方法 1：提取列值

要提取列表中指定列名的值，可以使用以下 extract_column 方法：

def extract_column(data, column_name):
    """
    提取列表中指定列名的值，并返回一个列表

    :param data: 包含字典的列表
    :param column_name: 要提取的列名
    :return: 包含提取值的列表
    """
    return [item[column_name] for item in data]

例如：

data = [
    {'id': 1, 'name': 'alice', 'age': 25},
    {'id': 2, 'name': 'bob', 'age': 30},
    {'id': 3, 'name': 'charlie', 'age': 35},
]

names = extract_column(data, 'name')
print(names)  # 输出：['alice', 'bob', 'charlie']

方法 2：提取键值对

如果需要将提取的列值作为键，可以使用以下 extract_column_to_dict 方法：

def extract_column_to_dict(data, key_column_name, value_column_name):
    """
    提取列表中指定键名和值名的字典

    :param data: 包含字典的列表
    :param key_column_name: 键名对应的列名
    :param value_column_name: 值对应的列名
    :return: 包含提取键值对的字典
    """
    return {item[key_column_name]: item[value_column_name] for item in data}

例如：

new_dict = extract_column_to_dict(data, 'id', 'name')
print(new_dict)  # 输出：{1: 'Alice', 2: 'Bob', 3: 'Charlie'}