【问题标题】:Convert ndarray to dict in python3在python3中将ndarray转换为dict
【发布时间】:2020-06-16 12:22:25
【问题描述】:

我有一个像这样的 ndarray

LABEL1              99       113           2010-04-26 20:12:23+00:00
LABEL1              29       143           2010-05-06 20:12:23+00:00
LABEL1              99       323           2010-02-12 20:12:23+00:00
LABEL1              23       223           2010-04-25 20:12:23+00:00
LABEL2              23        23           2010-01-21 20:12:23+00:00
LABEL1             234       123           2010-12-26 20:12:23+00:00
LABEL1              93       133           2010-02-23 20:12:23+00:00
LABEL4              19      1223           2010-07-24 20:12:23+00:00

我需要做一些聚合并返回为 dict..

我最后应该得到的类似于这个

[ 
  { 'LABEL1': { 'COLA':577,  'COLB': 1058, 'LAST': '2010-12-26 20:12:23+00:00' } },
  { 'LABEL2': { 'COLA':23,   'COLB': 23,   'LAST': '2010-01-21 20:12:23+00:00' } },
  { 'LABEL4': { 'COLA':19,   'COLB':1223,  'LAST': '2010-07-24 20:12:23+00:00' } }
]

我想的办法是转换成DataFrame,然后做一个group().agg...

aggr = select_df.groupby('LABELS').agg({'LABELS': [('LABELS', 'max')], 'COLA': [('COLA', 'sum'), ('COLB', 'count')], {'LAST': [('LAST', 'max')]})

我对 Python 有点陌生...并且对执行此操作所需的所有数据转换做噩梦...

原来的结构是一个列表

  [
    { 'Label': 'xxxx', 'LABELS': 'xxxx', 'COLA': ##, 'COLB': ##, 'LAST': 'datetime' },...
  ]

如果我可以简单地直接聚合这个列表,然后与下一个通道连接(列表以块的形式读取)以获得如上所述的最终列表......

【问题讨论】:

  • 你能分享print(your_array)的输出吗? 原来的结构是一个列表 你要把列表转换成一个ndarray,是吗?请提供正确的minimal reproducible example

标签: python pandas list dataframe arraylist


【解决方案1】:

您的尝试非常接近。

代码:

import pandas as pd

input = [
    {"LABELS": "LABEL1", "COLA": 99, "COLB": 113, "LAST": "2010-04-26 20:12:23+00:00"},
    {"LABELS": "LABEL1", "COLA": 29, "COLB": 143, "LAST": "2010-05-06 20:12:23+00:00"},
    {"LABELS": "LABEL1", "COLA": 99, "COLB": 323, "LAST": "2010-02-12 20:12:23+00:00"},
    {"LABELS": "LABEL1", "COLA": 23, "COLB": 223, "LAST": "2010-04-25 20:12:23+00:00"},
    {"LABELS": "LABEL2", "COLA": 23, "COLB": 23, "LAST": "2010-01-21 20:12:23+00:00"},
    {"LABELS": "LABEL1", "COLA": 234, "COLB": 123, "LAST": "2010-12-26 20:12:23+00:00"},
    {"LABELS": "LABEL1", "COLA": 93, "COLB": 133, "LAST": "2010-02-23 20:12:23+00:00"},
    {"LABELS": "LABEL4", "COLA": 19, "COLB": 1223, "LAST": "2010-07-24 20:12:23+00:00"},
]

df = (
    pd.DataFrame(input)
    .groupby(["LABELS"])
    .agg({"COLA": "sum", "COLB": "sum", "LAST": "max"})
)

print(df.to_dict("index"))

输出:

{'LABEL1': {'COLA': 577, 'COLB': 1058, 'LAST': '2010-12-26 20:12:23+00:00'}, 'LABEL2': {'COLA': 23, 'COLB': 23, 'LAST': '2010-01-21 20:12:23+00:00'}, 'LABEL4': {'COLA': 19, 'COLB': 1223, 'LAST': '2010-07-24 20:12:23+00:00'}}

【讨论】:

  • @AndreCouture 你能打印出你的输入吗?
【解决方案2】:

先转换成dataframe:

df:

    0       1   2   3
0   LABEL1  29  143 2010-05-06  20:12:23+00:00
1   LABEL1  99  323 2010-02-12  20:12:23+00:00
2   LABEL1  23  223 2010-04-25  20:12:23+00:00
3   LABEL2  23  23  2010-01-21  20:12:23+00:00
4   LABEL1  234 123 2010-12-26  20:12:23+00:00
5   LABEL1  93  133 2010-02-23  20:12:23+00:00
6   LABEL4  19  1223    2010-07-24  20:12:23+00:00

df.columns = ['label','x','y','z','w']

df.set_index('label').T.to_dict('dict')

结果:

{'LABEL1': {'x': 93, 'y': 133, 'z': '2010-02-23', 'w': '20:12:23+00:00'},
 'LABEL2': {'x': 23, 'y': 23, 'z': '2010-01-21', 'w': '20:12:23+00:00'},
 'LABEL4': {'x': 19, 'y': 1223, 'z': '2010-07-24', 'w': '20:12:23+00:00'}}

编辑: 然后 groupby 标签并按 sum 聚合,最大值

df.groupby(["label"])\
    .agg({"x": "sum", "y": "sum", "z": "max", "w": "max"}).T.to_dict('dict')

结果:

{'LABEL1': {'x': 478, 'y': 945, 'z': '2010-12-26', 'w': '20:12:23+00:00'},
 'LABEL2': {'x': 23, 'y': 23, 'z': '2010-01-21', 'w': '20:12:23+00:00'},
 'LABEL4': {'x': 19, 'y': 1223, 'z': '2010-07-24', 'w': '20:12:23+00:00'}}

【讨论】:

  • 差不多了..如何添加每组的所有值?现在它只显示一个值,如果我需要每组有 sum() ('label')。对于我需要最新的时间戳(最晚的时间)
猜你喜欢
  • 1970-01-01
  • 2017-05-16
  • 1970-01-01
  • 2023-03-22
  • 2019-10-31
  • 1970-01-01
  • 2012-10-15
  • 2021-08-09
  • 1970-01-01
相关资源
最近更新 更多