Pythonで配列の加工

Question

前提・実現したいこと

プログラミング初学者です。
pythonを使い、以下の配列aを加工して配列bのようにしたいです。
コードを書いてみましたが期待どうりになりません。
アドバイスをお願いします。

　　a = ['×', '×', '4', '1', '×', '×', '×', '2', '×', '8', '3', '1', '2', '×', '1']
　　b = ['41', '2', '8312', '1']
　　　配列aの値がxの場合、xは配列bへ入れない。
　　　配列aの値がx以外場合、配列bへ入れる。ただしx以外の値が連続した場合、結合して配列bへ入れる。

該当のソースコード

a = ['×', '×', '4', '1', '×', '×', '×', '2', '×', '8', '3', '1', '2', '×', '1']
b = []
n = (len(a))
i = 0
while i < n:
    if a[i] != '×':
        b.append(a[i])
        i += 1
    else:
        i += 1
print(b)

# このコードでは、 b = ['4', '1', '2', '8', '3', '1', '2', '1'] となり、
# 期待している b = ['41', '2', '8312', '1'] にならない。

Answer 1

xでないものが出てくるたびにbに入れるのでなく、一時的な保存場所を用意してそこで結合しておきます。
xが現れたときに、結合したものがあれば、それをbに入れるようにすればいいでしょう。

a = ['×', '×', '4', '1', '×', '×', '×', '2', '×', '8', '3', '1', '2', '×', '1']
b = []

stack = ''
for c in a:
    if c == '×':
        if stack != '':
            b.append(stack)
            stack = ''
    else:
        stack += c

print(b)

Comments

[edmond_h]

解決しました、ありがとうございました。
考え方参考になります。

Answer 2

a = ['×', '×', '4', '1', '×', '×', '×', '2', '×', '8', '3', '1', '2', '×', '1']
b = []
non_x=0

def add(i, non_x):
    if non_x != 0:
        b.append("".join(a[i-non_x:i]))

if a[-1] != "×":
    a.append("×")

for i in range(len(a)):
    print(i, non_x, a[i])
    if a[i] == "×":
        add(i, non_x)
        non_x=0
    else:
        non_x+=1

print(b)

こんな感じでいかがでしょう。

Comments

[edmond_h]

解決しました、ありがとうございました。
a[-1] ←こういう表記初めて知りました。

Answer 3

from functools import reduce
from operator import add
a = ['×', '×', '4', '1', '×', '×', '×', '2', '×', '8', '3', '1', '2', '×', '1']
b = [s for s in reduce(add, a).split('×') if s]
print(b)  # ['41', '2', '8312', '1']

Comments

[edmond_h]

解決しました、ありがとうございました。