標準入力をリスト（辞書）化したのち、特定のバリュー（max、minなど）のインデックスを表示するには

Question

データが標準入力として以下のように与えられます。
＃データの個数
5
＃データ
10　25
10　54
41　62
45　11
14　12

ここで、データの左の数字から右の数字の差が上位3番目までの
インデックスを表示したいのです。
以下のように、2次元リスト化して左の数字をc、右の数字をdとするとこまで
いったのですが、この後「差」を出していこうにも、インデックスの表記を
どうすればいいのかがわかりません。
enumerateメソッドを使うのか、indexメソッドを使うのか
いろいろ試したのですが、出力までこぎつけれません。
どなたか、ご教授下さい。

a= input().split()
A=int(a)
my_list=[]
for i in range(A):
my_list.append(input().split())
c = my_list[i][0]
d = my_list[0][i]

Answer 1

以下のように、2次元リスト化して左の数字をc、右の数字をdとするとこまで

いった

for i in range(A):
my_list.append(input().split())
c = my_list[i][0]
d = my_list[0][i]

いえ、できていないです。
実際に c と d の値を覗いてみると良いでしょう。

# a= input().split()

a= int(input())
A=int(a)
my_list=[]

for i in range(A):
    my_list.append(input().split())
    c = my_list[i][0]
    d = my_list[0][i]

    print(c, d)

実行結果 Wandbox

10 10
10 25
Traceback (most recent call last):
  File "prog.py", line 8, in <module>
    d = my_list[0][i]
IndexError: list index out of range

まずは差のリストを作ることから始めましょう。
次のようなリストを作れれば、問題の半分は解けたと言っても良いでしょう。

>>> diffs
[15, 44, 21, -34, -2]

Comments

[gyanonsh]

ありがとうございます。
いちおう、自分なりに出してみましたが、
もっと、すっきりできるはずだとおもいながら・・。

[LouiS0616]

実際には二行目でエラーが出るはずなのですが、全く同じコードを実行しているのでしょうか？

[gyanonsh]

すみません、よく teratailの使い方を理解しておらず、
自己回答欄に生意気にも投稿してしまいました・・。
以下の方法はどうでしょうか？

[gyanonsh]

すみません。これです
a= input()
A = int(a)
list_01 = []
list_02 = []
for i in range(A):
x1,y1=[int(i) for i in input().split()]
list_01.append(x1)
list_02.append(y1)
diff = [x-y for (x, y) in zip(list_01, list_02)]
index_max = diff.index(max(diff))+1
print(index_max)

[LouiS0616]

リストdiffを作ること及び最大値のインデックスの取得はループ後にした方が良いですね。

[gyanonsh]

a= input()
A = int(a)
list_01 = []
list_02 = []
for i in range(A):
x1,y1=[int(i) for i in input().split()]
list_01.append(x1)
list_02.append(y1)
diff = [x-y for (x, y) in zip(list_01, list_02)]
index_max = diff.index(max(diff))+1
print(index_max)

[LouiS0616]

コメント欄では行頭のスペースが削除されますので、質問を編集して追記する形にすると良いでしょう。

[gyanonsh]

なるほど。そのとおりですね。
いつもありがとうございます！

Answer 2

差のリストを先に作ってから、enumerateでインデックスと組みにして、sortedでkeyをうまく指定してソートすればよいかと思います。

追記

n = int(input())

diff_lst = []
for _ in range(n):
    a, b = map(int, input().split())
    diff_lst.append(a-b)

for i, d in sorted(enumerate(diff_lst), key=lambda x:x[1], reverse=True)[:3]:
    print(i)

Comments

[gyanonsh]

hayatakaさん
ありがとうございます。
色々な関数とかメソッドを使って、チャレンジするのですが、
それぞれの関数などが対応する型を理解していないので
なかなか、カチっとこないのが問題です。。
悔しいですねぇ・・。

[hayataka2049]

この手の問題でうまくいかないときはたいてい難しく考えすぎなので、とりあえずシンプルに考えることをおすすめします。

[gyanonsh]

＞難しく考えすぎ。。　最近そう思います。ありがとうございました！

Answer 3

A= int(input())
list_01 = []
list_02 = []
for i in range(A):
    x1,y1=[int(i) for i in input().split()]
    list_01.append(x1)
    list_02.append(y1)
    diff = [x-y for (x, y) in zip(list_01, list_02)]
    index_max = diff.index(max(diff))+1
print(index_max)

Answer 4

numpy を使うのがおすすめです。

手順

1. 行ごとに差の絶対値を計算する。(numpy.abs())
2. 昇順ソートする。(numpy.agrsort() で降順ソートし、[::-1] で反転させる。)
3. 差が大きい上位3つのインデックスを取り出す。([:3])

サンプルコード

import numpy as np

# 入力を受け付ける。
data = []
for i in range(2):
    values = input('空白区切りで2つの数字を入力してください。 (例: 1 5)').split()
    data.append(list(map(int, values)))
data = np.array(data)
print(data)

# 行ごとに差を計算する。
diff = np.abs(data[:, 0] - data[:, 1])
print(diff)  # [15 44 21 34  2]

# 昇順ソートする。
indices = np.argsort(diff)[::-1]
print(indices)  # [4 0 2 3 1]

for i, index in enumerate(indices):
    print('上位 {} 番目に差が大きい行 {} 値 {}'.format(i + 1, index, data[index]))
# 上位 1 番目に差が大きい行 1 値 [10 54]
# 上位 2 番目に差が大きい行 3 値 [45 11]
# 上位 3 番目に差が大きい行 2 値 [41 62]
# 上位 4 番目に差が大きい行 0 値 [10 25]
# 上位 5 番目に差が大きい行 4 値 [14 12]