複数のlistからfor文で最大値を取得して空のlistに入れたい

Question

python初心者です。
複数のlistからfor文を使って同じインデックスの最大値を空のlistに入れたいのですが、エラーが出てしまいます。

IndexError: invalid index to scalar variable.

list1 = [7, 4, 3]
list2 = [2, 1, 8]
list3 = [9, 3, 1]

list0 = []

各listの同じインデックスの要素の最大値をとってlist1が最大ならlist0に1をいれ、
同様にlist2がならlist0に2、list3が最大ならlist0に3という風にしたいです。

求めている結果は下のlist0の様なものです。
list0 = [3, 1, 2]

for i in range(0,3):
    for j in range(0,3):
        if list1[i][j] > list2[i][j] and list1[i][j] > list3[i][j]:
            list0[i][j] = 1
        else:
            pass
for i in range(0,3):
    for j in range(0,3):
        if list2[i][j] > list1[i][j] and list2[i][j] > list3[i][j]:
            list0[i][j] = 2
        else:
            pass
for i in range(0,3):
    for j in range(0,3):
        if list3[i][j] > list1[i][j] and list3[i][j] > list2[i][j]:
            list0[i][j] = 3
        else:
            pass

Answer 1

各列における最大値は１つである前提の回答です。

• zip([～])で各列ずつ、値の組（タプル）tを得ます。
• max(v)にて値の組の最大値を得てv.index(～)で最大値に一致する、値の組のインデックス位置を得ます。

list1 = [7, 4, 3]
list2 = [2, 1, 8]
list3 = [9, 3, 1]
print([t.index(max(t)) + 1 for t in zip(list1, list2, list3)]) # [3, 1, 2]

Comments

[yonhark0]

助かりました、ありがとうございます！

Answer 2

こんにちは。
参考までに、自分ならこういうコードを書きます、というのを回答します。リスト内の最大値の要素のインデクスを得るために、 numpy.argmax を使いました。

import numpy as np

list1 = [7, 4, 3]
list2 = [2, 1, 8]
list3 = [9, 3, 1]

list0 = [1+np.argmax(t) for t in zip(list1, list2, list3)]

• 動作確認用CodePen： https://repl.it/@jun68ykt/Q239764

　

• 参考： Qiita リストの最大値（最小値）のindexを取得する6つの方法を計測してみた (kasyunさん）

Comments

[yonhark0]

ありがとうございます！

Answer 3

やりたいのはこうじゃないかな？

list1 = [7, 4, 3]
list2 = [2, 1, 8]
list3 = [9, 3, 1]
list0 = [0] * 3
for column in range(3):
    if list2[column] <= list1[column] >= list3[column]:
        list0[column] = 1
for column in range(3):
    if list1[column] <= list2[column] >= list3[column]:
        list0[column] = 2
for column in range(3):
    if list1[column] <= list3[column] >= list2[column]:
        list0[column] = 3
print(list0)

もっと簡単にできますよ。

list0 = [column.index(max(column)) + 1 for column in zip(list1, list2, list3)] 

Comments

[yonhark0]

助かりました、ありがとうございます！