.unique()与to_dic()结合

在 Pandas 中,unique() 方法用于获取 Series 或 DataFrame 中唯一值的数组。当你使用 groupby 方法对 DataFrame 进行分组后,你可以在分组结果上调用 unique() 方法来获取每个分组中唯一值的数组。这个方法返回的结果可以直接转换为字典。

让我详细说明一下:

假设 price_df 是一个 DataFrame,其中包含了商品的价格信息。你想要按照 "itemId" 列进行分组,并且获取每个分组中的 "discountePrice" 列的唯一值数组。你可以按照以下步骤操作:

python 复制代码
import pandas as pd

# 假设price_df是你的DataFrame
# 创建一个示例DataFrame
data = {
    'itemId': ['A', 'A', 'B', 'B', 'B'],
    'discountePrice': [10, 15, 20, 25, 20]
}
price_df = pd.DataFrame(data)

# 按照 'itemId' 进行分组,并获取 'discountePrice' 列的唯一值数组
unique_discount_prices = price_df.groupby('itemId')['discountePrice'].unique()

# 将结果转换为字典
result_dict = unique_discount_prices.to_dict()

print("得到的字典:")
print(result_dict)

这将输出:

css 复制代码
得到的字典:
{'A': array([10, 15]), 'B': array([20, 25])}

这里,unique_discount_prices 是一个 Series,其索引是 "itemId",值是包含唯一 "discountePrice" 值的数组。调用 to_dict() 方法将 Series 转换为字典,其中键是 "itemId",值是对应的唯一值数组。

问题1:

如果要得到所有值的分组,没不是唯一值的分组,应该怎么实现?

答:用.apply(list)方法

all_discount_prices = price_df.groupby('itemId')'discountePrice'.apply(list)

输出结果: 这将输出一个字典,其中键是 "itemId",值是包含相应分组中所有 "discountePrice" 值的列表:

less 复制代码
lessCopy code
得到的字典:
{'A': [10, 15], 'B': [20, 25, 20]}

在这个字典中,每个键对应一个 "itemId",值是一个包含相应分组中所有 "discountePrice" 值的列表。

相关推荐
星云穿梭5 小时前
用Python写一个带图形界面的学生管理系统——完整教程
python
金銀銅鐵5 小时前
用 Pygame 实现 15 puzzle
python·数学·游戏
黄忠11 小时前
大模型之LangGraph技术体系
python·llm
hboot1 天前
AI工程师第二课 - 数据处理
人工智能·python·数据分析
用户8356290780511 天前
使用 Python 自动化 PowerPoint 形状布局与格式设置
后端·python
用户8356290780511 天前
用 Python 自动化 PowerPoint 演讲者备注添加
后端·python
黄忠1 天前
01-系统架构设计-LangGraph状态机与多源异构RAG
python
zzzzzz3101 天前
假如我是掘金管理员,我先给评论区装个'代码审查'系统
python·程序员·机器人
砍材农夫2 天前
python环境|conda安装和使用(2)
后端·python
程序员龙叔2 天前
编写高质量 Skill 系列 -- 如何设计需求分析与用例生成的 SKILL
自动化测试·软件测试·python·软件测试工程师·接口测试·性能测试·skill·ai测试