作者cuteSquirrel (松鼠)
标题Re: [问题] 列表中依据元素分组
时间Wed Dec 18 17:03:52 2019
※ 引述《radiant77 (七七)》之铭言:
有一个列表,长度非固定,其中有A、B、C、D、NAV等字
希望依据总分放到不同的list中,只看英文字母,前面的数字不算分
A = 1分
B = 2分
C = 2分
D = 3分
NAV或0分的一组
例如:'7A-1A'有两个A,就是1+1=2分,'1C', '3B'也都是2分
'1A-7C-3A',就是1+2+1=4分,'28A-7A-3A-1A'是1+1+1+1=4分
------------
这项需求的关键在 计分。
比较好维护的写法是把计分拉出来写成一个function。
之後若修改计分规则(例如ABCD权重改变 或者外加别的特殊条件),
只要计分的function随着修正即可。
可用工具:dictionary, filter, lambda
参考范例:
https://www.onlinegdb.com/rJvuhKP0S
# compute score for input string
def get_score( s:str )->int:
if 'NAV' in s:
return 0
grade_table = { 'A': 1, 'B': 2, 'C': 2, 'D' : 3}
score = 0
for idx, grade in enumerate(grade_table.keys()):
score += s.count( grade ) * grade_table[grade]
return score
if __name__ == '__main__':
list_test = ['1A', '7A', ..., 为节约版面,後续元素和原本的叙述相同 ]
score_group = [None] * 5
# create score group from 0 to 4
for wanted_score in range(0,5):
score_group[wanted_score] \
= list( filter (lambda x: (get_score(x) == wanted_score), list_test) )
# output score group from 0 to 4
for s, g in enumerate(score_group):
print( "{score} score list: {group}".format( score = s, group = g) )
※ 编辑: cuteSquirrel (101.12.103.116 台湾), 12/18/2019 21:20:09
1F:推 radiant77: 我的python2.7第一行会出错,不过还是谢谢指导 12/19 14:53
2F:→ cuteSquirrel: 把type hint 的语法拿掉就可以在 python 2.7 跑了 12/19 16:01
5F:→ cuteSquirrel: :) 12/19 17:39