python - 将值分组到字典中的列表(Grouping values to a list in dict)

Question

While reading a csv file using csv.DictReader

(使用csv.DictReader读取csv文件时)

I get

(我懂了)

[{'id': 1, 'status1': '1', 'status2': '2', 'status3': '3' }]

How can I manuplate while reading or later to get:

(在阅读或以后阅读时如何处理：)

[{'id': 1, 'status': ['1', '2', '3']}]

TLDR;

(TLDR；)

I want to group similar fields into a list.

(我想将类似的字段分组到一个列表中。)

and/or - how can i do this in pandas pd.read_csv() too?

(和/或-我也如何在pd.read_csv（）中做到这一点？)

Thanks in advance!

(提前致谢！)

  ask by sgp translate from so

Answer 1

If it is certain that the only fields you want to group are those who end with digits, you can use a regex to identify them, and append their corresponding values to a list:

(如果确定要分组的唯一字段是以数字结尾的字段，则可以使用正则表达式来识别它们，并将其对应的值附加到列表中：)

import re

def compact_dict(d):
    compact = {}
    for key, value in d.items():
        # a group containing anything, followed by at least one digit
        m = re.match(r'(.*)d+', key)  
        if m:
            # the key that we use is the original one without the final digits
            compact.setdefault(m.group(1), []).append(value)
        else:
            # not part of a group of similar keys, we just store the key:value pair
            compact[key] = value
    return compact

data = [{'id': 1, 'status1': '1', 'status2': '2', 'status3': '3' }]
out = [compact_dict(d) for d in data]
print(out)
# [{'id': 1, 'status': ['1', '2', '3']}]