如何在Python中实现类似PHP array_column的方法来处理嵌套列表？

Python中处理嵌套列表，提取特定字段值的需求很常见，这与PHP的array_column函数功能类似。本文将演示如何用Python优雅地实现类似功能，并提供更规范的代码示例。

假设我们有一个包含多个字典的列表：

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

我们需要提取特定字段的值，或者以一个字段为键，另一个字段为值，创建新的字典。

为此，我们定义两个函数：extract_column 和 extract_column_to_dict。

立即学习“PHP免费学习笔记（深入）”；

Moshi Chat

法国AI实验室Kyutai推出的端到端实时多模态AI语音模型，具备听、说、看的能力，不仅可以实时收听，还能进行自然对话。

下载

extract_column 函数用于提取指定列名的所有值，返回一个列表：

def extract_column(data, column_name):
    """
    从字典列表中提取指定列名的值。

    Args:
        data: 字典列表。
        column_name: 要提取的列名。

    Returns:
        包含提取值的列表。  如果列表为空或列名不存在，返回空列表。
    """
    if not data:
        return []
    try:
        return [item[column_name] for item in data]
    except KeyError:
        return []

extract_column_to_dict 函数则将指定列名作为键，另一列名作为值，构建一个字典：

def extract_column_to_dict(data, key_column_name, value_column_name):
    """
    从字典列表中提取指定列名作为键，另一列名作为值，创建字典。

    Args:
        data: 字典列表。
        key_column_name: 作为键的列名。
        value_column_name: 作为值的列名。

    Returns:
        包含键值对的字典。如果列表为空或列名不存在，返回空字典。
    """
    if not data:
        return {}
    try:
        return {item[key_column_name]: item[value_column_name] for item in data}
    except KeyError:
        return {}

使用方法如下：

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']

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

#处理不存在的列名
empty_list = extract_column(data, 'nonexistent')
print(empty_list) #输出: []

empty_dict = extract_column_to_dict(data, 'id', 'nonexistent')
print(empty_dict) #输出: {}

这两个函数提供了更健壮、更易读的Python实现，有效地模拟了PHP array_column的功能，并增加了错误处理，避免了 KeyError 异常。