リスト内全体に追加される数値を足す方法について

Question

前提・実現したいこと

野球の点数のプログラムを作ろうとしているのですが、リスト内全体に追加される数値を足す方法が
思考錯誤しても出来ません。リスト2つ目までは上手くいくのですが...。
（現状）ヒット、ヒット、ヒットの場合リストに格納される数[1,2,2]
例　ヒット、ヒット、ヒットの場合リストに格納される数は[1,2,3]にしたい。

該当のソースコード

#野球ポイントプログラム
#set_pointリストはヒットを格納
set_point =[]
con_set = []
#numはヒットで１、ツーベースで２、スリーベースで３
#num2はホームラン
def num_point(num,num2):
    point = 0
    if num2 == 0:
        set_point.append(num)
        new_set = list(map(lambda x: x+num, set_point[:-1]))
        new_set.append(num)
#関数外でないといけ無いのでcon_setに１度格納
        for row in new_set:
            con_set.append(row)
            for row in con_set:
#リスト内から4以上の数字を消し、消した数をpointに点数
                 if row >= 4:
                    con_set.remove(row)
                    point += 1
                    return point
                 else:
                    return point
#ホームランの場合
    elif num2 >= 1:
        point = len(con_set) + point + 1
        if len(con_set) >= 1:
            set_point.clear()
            con_set.clear()
            return point
        else:
            return point



print(num_point(1,0))
print(con_set)

補足情報（FW/ツールのバージョンなど）

python3.7

Comments

[LouiS0616]

例えば三本連続して単打が出たときset_pointは[1, 1, 1]になっていますが、これは想定通りですか？

[HaTea]

set_pointのリストは想定通りです。
com_setのリストで処理しようとしています。

Answer 1

set_pointのリストは想定通りです。

なるほど、それならば間違っている箇所がはっきり分かります。

new_set = list(map(lambda x: x+num, set_point[:-1]))

[1, 1, 1] を [1, 2, 3] にしたいならば、各要素の累積和を出さねばなりません。
しかし上記のコードでは、各要素にnumを足しているだけです。

累積和は例えばitertools.accumulateを用いて求めることができます。

import itertools

src = [1, 1, 1]
dst = list(itertools.accumulate(src))
print(dst)  # => [1, 2, 3]

実際に差し替えて試してみて下さい。

組んでみた

組んでみました。
雑ですけれど、グローバル変数を利用しないで実現できる例です。参考までに。

# bは塁打。ホームランも4塁打として扱えば良い。
def hit(base_state, b):
    # どの塁までランナーが到達したか。3塁のあとは4塁,5塁と突き進む。
    new_state = [b]
    for e in base_state:
        new_state.append(e+b)
        
    # print(new_state)
         
    # 4塁を超えたランナーをホームインさせる。
    ret = []
    point = 0
    for e in new_state:
        if e < 4:
            ret.append(e)
        else:
            point += 1
            
    # ベース上の状況と、今の打席の点数を返す。
    return ret, point
   

# 2塁打、単打、3塁打、単打、単打、本塁打、本塁打、2塁打
h = [2, 1, 3, 1, 1, 4, 4, 2]
# 最初はランナーなし。
state = []

for e in h:
    print('-'*16)
    if not state:
        print('ランナーなし')
    else:
        print(f'ただいまランナー{state}塁')
        
    print(f'{e}塁打!')
    state, point = hit(state, e)    # 塁上の状況の更新。
    print(f'{point}点入りました')
    
print(f'残塁は{state}です。')

実行結果 Wandbox

----------------
ランナーなし
2塁打!
0点入りました
----------------
ただいまランナー[2]塁
1塁打!
0点入りました
----------------
ただいまランナー[1, 3]塁
3塁打!
2点入りました
----------------
ただいまランナー[3]塁
1塁打!
1点入りました
----------------
ただいまランナー[1]塁
1塁打!
0点入りました
----------------
ただいまランナー[1, 2]塁
4塁打!
3点入りました
----------------
ランナーなし
4塁打!
1点入りました
----------------
ランナーなし
2塁打!
0点入りました
残塁は[2]です。

Comments

[HaTea]

累積和の場合、単打→二塁打→三塁打の場合リストは[1, 3, 6]になってしまい4以上になるのは1つだけなのでpointは１になってしまいます。
上記の場合、実現したいのは[1+2+3, 2+3, 3] の様な[6, 5, 3]形です。

[HaTea]

すみません。最後まで見ていませんでした。
上のQは消しておきます。

即日に回答して下さりありがとうございました。
とても助かりました！