Question

如何以下表输出格式写下面的数据？

dict列表

data=[{'date': datetime.date(2011, 2, 8), 'user': u'xxx', 'status': u'P'}, {'date': datetime.date(2011, 2, 8), 'user': u'yyy', 'status': u'P'}, {'date': datetime.date(2011, 2, 8), 'user': u'zzz', 'status': u'P'}, {'date': datetime.date(2011, 2, 9), 'user': u'xxx, 'status': u'P'}, {'date': datetime.date(2011, 2, 9), 'user': u'yyy', 'status': u'E'}, {'date': datetime.date(2011, 2, 9), 'user': u'zzz', 'status': u'E'}, {'date': datetime.date(2011, 2, 10), 'user': u'xxx', 'status': u'P'}, {'date': datetime.date(2011, 2, 10), 'user': u'yyy', 'status': u'P'}, {'date': datetime.date(2011, 2, 10), 'user': u'zzz', 'status': u'P'}]

输出格式应为：

S.no  user  2011-02-08 2011-02-09 2011-02-10  p-total E-total total 
 1    xxx      p          p         p           3       0       3
 2    yyy      p          E         p           2       1       3
 3    zzz      p          E         E           1       2       3

请帮助

Answer 1

我对写这样的答案感到有些矛盾 - 似乎只是制作了一个完整的解决方案而且教学价值非常小，但是我试图让它尽可能地有用。 的

如果我了解您正在尝试正确执行的操作，则需要将data转换为CSV，其中每个用户都有一行。有一系列日期，您希望每个日期有一列 - 该列表示该日期用户的状态。然后有列生成每个日期的每个状态的总计，依此类推。您引用的输出看起来最像CSV，其中标签为分隔符，但正如eumiro指出的那样，它并不完全正确。但是，我们假设您要编写以制表符分隔的数据。如果您在data中发现用户在一天内有两种不同的状态，那么您的问题就不清楚会发生什么情况，所以让我们检查并抛出异常。

请注意，最后一段中的所有内容都应该出现在您的问题中，以及目前为止您最佳尝试的代码。

因此，使用csv模块中的DictWriter是一个合理的想法，但要使用该类，每行需要一个字典，将列标题映射到值。因此，您可以迭代data中的所有内容以生成字典字典，将用户映射到表示该用户的行的字典。你可以这样做：

from collections import defaultdict
import csv
from datetime import date

user_to_row = defaultdict(dict)

for d in data:
    user = d['user']
    status = d['status']
    row_dict = user_to_row[user]
    row_dict['user'] = user
    date_string = str(d['date'])
    if date_string in d and row_dict[date_string] != status:
        raise Exception, "Contradiction: '%s' on '%s'" % (user,date_string)
    row_dict[date_string] = status
    # If a value isn't set in one of the total columns yet, set it to 0:
    row_dict.setdefault('p-total',0)
    row_dict.setdefault('E-total',0)
    row_dict.setdefault('total',0)
    # Make sure you increment the right column:
    count_column = 'p-total' if (status == 'P') else 'E-total'
    row_dict[count_column] += 1
    # And increment the overall total column in any case:
    row_dict['total'] += 1

您应该检查一下您是否了解其中发生了什么 - 尝试打印user_to_row以检查您是否了解正在制作的内容。

现在您只需要遍历user_to_row字典中的值并使用DictWriter输出它们。这里需要注意的是，您不确定每个日期都会有一个条目，所以在这种情况下我只是在缺少值时插入Unknown：

with open("hello.csv","w") as f:

    # Create the headings:
    headings = ['S.no']
    headings += [str(date(2011,2,i)) for i in xrange(6,11)]
    headings += ['user', 'date_format','p-total','E-total','total']

    writer = csv.DictWriter(f, headings, delimiter="\t")

    # The writeheader method only appeared in Python 2.7, so write the
    # headings from a dictionary that maps each heading to itself:
    writer.writerow(dict(zip(headings,headings)))

    # Assume that S.no is just a row number...
    sno = 1
    for d in user_to_row.values():
        d['S.no'] = sno
        # Fill in any unknown values with 'Unknown':
        for h in headings:
            d.setdefault(h,'Unknown')
        writer.writerow(d)
        sno += 1

csv module的文档应该为您提供了解该部分所需的所有额外信息。

然后输出如下：

S.no    2011-02-06  2011-02-07  2011-02-08  2011-02-09  2011-02-10  user    date_format p-total E-total total
1   Unknown Unknown P   P   P   xxx Unknown 3   0   3
2   Unknown Unknown P   E   P   yyy Unknown 2   1   3
3   Unknown Unknown P   E   P   zzz Unknown 2   1   3

...因为标签而在这里看起来很奇怪，但会正确加载到电子表格中。

如何以表格格式编写dict数据

1 个答案: