PHP前端开发

Python如何模拟PHP的array_column函数?

百变鹏仔 1个月前 (12-15) #PHP
文章标签 函数

使用 python 模拟 php array_column 方法

在 php 中,array_column 方法可用于从多维数组中提取指定列的值。本文将介绍如何使用 python 模拟该方法。

为了实现类似 php 中的 array_column 的功能,可以将数据封装成两个方法:

def extract_column(data, column_name):    """    提取列表中指定列名的值,并返回一个列表    :param data: 包含字典的列表    :param column_name: 要提取的列名    :return: 包含提取值的列表    """    return [item[column_name] for item in data]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}

以下是如何使用这些方法提取指定列的值:

立即学习“PHP免费学习笔记(深入)”;

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']new_dict = extract_column_to_dict(data, 'id', 'name')print(new_dict)  # 输出:{1: 'Alice', 2: 'Bob', 3: 'Charlie'}