リストの長さが合わない

リストの長さが合わないです。

#coding:utf-8

A = []
B = []

data =[[["Where","is","Mary"],["Bathroom","is"]],[["Bob","is","23"],["Oh"]]]

for i in range(len(data)):
        tmp1 = []
        for j in range(len(data[i])):
            for k in range(len(data[i][j])):
                    tmp1.append(data[i][j][k])
            A.append(tmp1)

for i in range(len(data)):
        tmp2 = []
        for j in range(len(data[i])):
            if j == 0:
                for k in range(len(data[i][j])):
                        tmp2.append(data[i][j][k])
                B.append(tmp2)

print(len(A))
print(len(B))

とコードを書いて実行すると、

4
2

と出力されました。
しかし、理想のアウトプットはAとBの長さを２にしたいです。なので、Aが余計にforループ回っていることになります。
長さの２はどこからきているかというと、

[["Where","is","Mary"],["Bathroom","is"]]

や

[["Bob","is","23"],["Oh"]]

の配列の長さが２であることからきています。
なのでtmp1には、まず

["Where","is","Mary","Bathroom","is"]

が入って、次にtmp1に

["Bob","is","23","Oh"]

が入り、Aには

[["Where","is","Mary","Bathroom","is"],["Bob","is","23","Oh"]]

という形にしたいです。

tmp2には、まず

["Where","is","Mary"]

が入って、次にtmp2に

["Bob","is","23"]

が入り、Bには

[["Where","is","Mary"],["Bob","is","23"]]

という形にしたいです。

なぜ、意図した挙動にならないのでしょうか？

LouiS0616

2018/08/15 09:10

Aの値およびBの値がどのようになるのが理想なのか、端的に書いてください。

k0908

2018/08/15 09:13

Aは、[["Where","is","Mary","Bathroom","is"],["Bob","is","23","Oh"]]　となり、Bは[["Where","is","Mary"],["Bob","is","23"]]　となるようにしたいです

行動規範の内容に同意します

回答1件

ベストアンサー

なぜ、意図した挙動にならないのでしょうか？

Aにtmp1を加えるタイミングが変です。次のように書けば問題なく動作します。

Python
1for i in range(len(data)):
2    tmp1 = []
3    for j in range(len(data[i])):
4        for k in range(len(data[i][j])):
5            tmp1.append(data[i][j][k])
6    A.append(tmp1)

正直ご提示のコードの処理は面倒で追っていないのですが、
Aの値が次のようになっていたことから解決策はすぐに見当がつきました。

Python
1[['Where', 'is', 'Mary', 'Bathroom', 'is'], ['Where', 'is', 'Mary', 'Bathroom', 'is'], ['Bob', 'is', '23', 'Oh'], ['Bob', 'is', '23', 'Oh']]

わざわざlenして潰した情報を観察しても意味が無いです。
リスト長が合わないのは、リストが理想と離れている副次的な影響に過ぎません。

別解

Aは、[["Where","is","Mary","Bathroom","is"],["Bob","is","23","Oh"]]　となり、Bは[["Where","is","Mary"],["Bob","is","23"]]　となるようにしたいです

それならば、難しく考えすぎでは無いでしょうか。

Python
1data = [[["Where","is","Mary"],["Bathroom","is"]],[["Bob","is","23"],["Oh"]]]
2
3A = [sum(inner, []) for inner in data]
4# A = [functools.reduce(lambda l1, l2: l1 + l2, inner) for inner in data] でも可
5# A = [list(itertools.chain(*inner)) for inner in data] でも可
6
7B = [e[0] for e in data]
8# B = [ef for ef, *_ in data] でも可
9
10print(len(A), A)  # 2 [['Where', 'is', 'Mary', 'Bathroom', 'is'], ['Bob', 'is', '23', 'Oh']]
11print(len(B), B)  # 2 [['Where', 'is', 'Mary'], ['Bob', 'is', '23']]