📣 独立日限时特惠: VIP通行证立减$68
查看: 1132| 回复: 2
跳转到指定楼层
上一主题 下一主题
收起左侧

[学Python/Perl] 求问一道python题

全局:

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看附件。没有帐号?注册账号

x
已知一周七天的log
log = {car:[0,1,0,0,0,1,0],bus:[1,0,0,0,0,0,1],walk:[0,0,1,0,0,0,0]}
group = {overall:[car, bus, walk], vehicle:[car, bus]}
得结果 => {overall:[1,1,1,0,0,1,1], vehicle: [1,1,0,0,0,0,1]}

先谢谢各位大神,请问如何用最少的for loop做这道题。比方说什么zip(), any().

上一篇:怎样写好sql?
下一篇:关于转码路线和未来的一些疑问?真诚求各位前辈的意见!
推荐
magicsets 2021-2-3 17:38:56 | 只看该作者
全局:
  1. log = {
  2.     "car": [0, 1, 0, 0, 0, 1, 0],
  3.     "bus": [1, 0, 0, 0, 0, 0, 1],
  4.     "walk": [0, 0, 1, 0, 0, 0, 0]
  5. }
  6. group = {
  7.     "overall": ["car", "bus", "walk"],
  8.     "vehicle": ["car", "bus"]
  9. }

  10. from functools import reduce
  11. from operator import __or__


  12. result = {
  13.     group_name: reduce(lambda values, label: map(__or__, values, log[label]),
  14.                        labels, [0] * 7)
  15.     for group_name, labels in group.items()
  16. }
复制代码

补充内容 (2021-2-4 04:10):
用__or__是把log里的0/1理解成表示存在性的bool类型了,如果求和的话把__or__替换成__add__就可以
回复

使用道具 举报

全局:
不管怎么写都是O(n)的复杂度 本质上都是三个数组遍历一遍 用map甚至可以不出现for loop

vehicle = list(map(sum, zip(car, bus)))
overall = list(map(sum, zip(vehicle, walk)))
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册账号
隐私提醒:
  • ☑ 禁止发布广告,拉群,贴个人联系方式:找人请去🔗同学同事飞友,拉群请去🔗拉群结伴,广告请去🔗跳蚤市场,和 🔗租房广告|找室友
  • ☑ 论坛内容在发帖 30 分钟内可以编辑,过后则不能删帖。为防止被骚扰甚至人肉,不要公开留微信等联系方式,如有需求请以论坛私信方式发送。
  • ☑ 干货版块可免费使用 🔗超级匿名:面经(美国面经、中国面经、数科面经、PM面经),抖包袱(美国、中国)和录取汇报、定位选校版
  • ☑ 查阅全站 🔗各种匿名方法

本版积分规则

>
快速回复 返回顶部 返回列表