二次元配列の要素を一つずつ読み込んで処理したい

Question

二次元配列Bの要素を一つずつ読み込んで以下の処理を行いたいです。
書き方が分からない行は疑問点１、疑問点２としてコメントアウトしてあります。

A = [['apple', 1],['banana', 2],['orange', 3],['grape', 4]]
B = [['apple', 'ringo'],['banana', 'banana'],['orange', 'orenji'],['grape', 'budou']]

#(疑問点１)Bの０番目の要素（['apple', 'ringo']）をCとして取り出す方法
#C = ['apple', 'ringo']

result = []
while C:
        for m in range(len(A)):
            if A[m][0] == C[0]:
                a = []
                a.append(A[m]) 
                a.append(C[1]) 
                
                 result.append(a) 
                #（疑問点２）Aの１番目の要素を読み込む式

上のような処理をして以下の結果を得たいと考えています。

result = [['apple', 1, 'ringo'],['banana', 2, 'banana'],['orange', 3, 'orenji'],['grape', 4, 'budou']]

プログラムの中に書き方が分からない箇所が２つ（疑問点１と疑問点２）あります。
どのように書いたら良いかご教示ください。

Answer 1

以下でどうでしょう？

result = [x[0]+[x[1][1]] for x in zip(A, B)]

- **動作確認用 Repl.it: **https://repl.it/@jun68ykt/Q230905

※質問者さんからのコメントに

実際のプログラムではBのデータの[m][0]にAにはない要素がある

とあったので、上記の回答は取り下げます。

Comments

[nijihan]

実際のプログラムではBのデータの[m][0]にAにはない要素があるため、
if A[m][0] == C[0]:
の処理をつかいたいです。質問で書いたプログラムに#の部分の式を追加する方法では改善できないのでしょうか。

[jun68ykt]

@nijihanさん

> 実際のプログラムではBのデータの[m][0]にAにはない要素がある

ということなのですね。だとすると、上記の回答に挙げたコードは使えないですのでいったんスルーしてください。

> 質問で書いたプログラムに#の部分の式を追加する方法

については、検討してよい方法が思いつけば回答に追記します。

[nijihan]

readlineのリスト版のような、一つだけリストの要素を読み込む方法があればいいのですが…

Answer 2

A = [['apple', 1],['banana', 2],['orange', 3],['grape', 4]]
B = [['apple', 'ringo'],['banana', 'banana'],['orange', 'orenji'],['grape', 'budou']]

result = [[*a, b[1]] for b in B for a in A if a[0] == b[0]]
print(result)

データが大量にあるなら、二重ループは時間がかかり過ぎるので、辞書化した方がいいでしょう。

A = [['apple', 1],['banana', 2],['orange', 3],['grape', 4],['x',0]]
B = [['apple', 'ringo'],['banana', 'banana'],['orange', 'orenji'],['grape', 'budou']]

a = {a[0]: a for a in A}
result = [[*a[b0], b1] for b0, b1 in B if b0 in a]
print(result)

お書きになったコードで対応しました。

A = [['apple', 1],['banana', 2],['orange', 3],['grape', 4]]
B = [['apple', 'ringo'],['banana', 'banana'],['orange', 'orenji'],['grape', 'budou']]

result = []
for b in B:
    for a in A:
        if a[0] == b[0]:
             result.append([*a, b[1]])

print(result)

Comments

[shiracamus]

対応しました。