如何按列中的两个输入值分组并在

Question

电流输入：

reg1 test1 f day1
reg1 test2 p day1
reg1 test3 p day2
reg1 test2 p day2
reg1 test2 p day3
reg1 test4 f day3

期望输出：

reg   test  day1 day2 day3
reg1  test1   f   -    -
reg1  test2   p   p    p
reg1  test3   -   p    -
reg1  test4   -   -    f

如何从当前输入中获得所需的输出？我只想使用python的内置特性，因为我无法下载pandas、tablate等库

Answer 1

假设你的数据在stuff.csv。从加载开始

tups = [l.strip().split(' ') for l in open('stuff.csv') if l.strip()]
>>> tups
[['reg1', 'test1', 'f', 'day1'],
 ['reg1', 'test2', 'p', 'day1'],
 ['reg1', 'test3', 'p', 'day2'],
 ['reg1', 'test2', 'p', 'day2'],
 ['reg1', 'test2', 'p', 'day3'],
 ['reg1', 'test4', 'f', 'day3']]

现在找到所有可能的日子

days = sorted(set(tup[3] for tup in tups))
>>> days
{'day1', 'day2', 'day3'}

现在计算regs和tests的所有条目

from collections import defaultdict

d = defaultdict(dict)
for tup in tups:
    d[(tup[0], tup[1])][tup[3]] = tup[2]
>>> d
defaultdict(dict,
            {('reg1', 'test1'): {'day1': 'f'},
             ('reg1', 'test2'): {'day1': 'p', 'day2': 'p', 'day3': 'p'},
             ('reg1', 'test3'): {'day2': 'p'},
             ('reg1', 'test4'): {'day3': 'f'}})

最后，计算输出列表：

out = []
for e in d:
    out.append(
        list(e) + [d[e][day] if day in d[e] else '-' for day in days])
>>> out
[['reg1', 'test2', 'p', 'p', 'p'],
 ['reg1', 'test4', '-', 'f', '-'],
 ['reg1', 'test1', '-', '-', 'f'],
 ['reg1', 'test3', 'p', '-', '-']]

在CSV中输出这个应该很简单。你知道吗