如何以下表输出格式写入下表数据？
字典列表
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
需要帮助
请您参考如下方法:
我对写这样的答案有点矛盾 - 它似乎只是提供了一个完整的解决方案，几乎没有教学值(value)，但我已经尽力让它尽可能地有用......
如果我理解您正在尝试正确执行的操作，您希望将您的 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
...由于选项卡，这里看起来很奇怪，但会正确加载到电子表格中。
