list index out of rangeのエラーについて

Question

前提・実現したいこと

n×m行列を入力し、各行と列の合計を挿入した新しい表を出力するプログラムを作成しています。

発生している問題・エラーメッセージ

各行と列の合計を求めて挿入する段階で下記のエラーが起きてしまいます。

IndexError                                Traceback (most recent call last)
<ipython-input-7-67709c5dbd4f> in <module>()
     16 for i in range(n):
     17     for j in range(m):
---> 18         A[i][m+1] += A[i][j]
     19 
     20 for j in range(m+1):

IndexError: list index out of range

該当のソースコード

n, m = map(int, input().split())

A = [[0]*(m+1) for i in range(n+1)]

for i in range(n):
    B = list(map(int, input().split()))
    for j in range(m):
        A[i][j] = B[j]
    
for i in range(n):
    for j in range(m):
        A[i][m+1] += A[i][j]

for j in range(m+1):
    for i in range(n):
        A[n+1][j] += A[i][j]
        
for i in range(n+1):
    print(A[i][0], end='')
    for j in range(1, m+1):
        print(' %d'%A[i][j], end='')
    print()

試したこと

最初の入力段階で直接行列Aに入力してしまうと行数が書き換えられてしまうとのことだったので新たに行列Bを作成し、そちらに入力してから行列Aに代入しなおす仕組みに書き換えてみたのですが、それでも行数が書き換えられてしまっているようでうまく出力が行えません。

補足情報（FW/ツールのバージョンなど）

python 3.6.2
ここにより詳細な情報を記載してください。

Comments

[hayataka2049]

3*4くらいの小さい行列の入力例、それを入力した際に期待される出力を教えてください。

[chibiyt]

回答ありがとうございます

http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ITP1_7_C&lang=jp

こちらの問題にある入力例と出力例になります。

Answer 1

とりあえず、インデックスの番号は0から始まってn-1（n:要素数）で終わるので、サイズを縦横1つ増やしたとしてもm+1とかは取れません。mまでです（縦横1つずつ増やさなければm-1まで）。

それだけ修正したコード。

n, m = map(int, input().split())

A = [[0]*(m+1) for i in range(n+1)]

for i in range(n):
    B = list(map(int, input().split()))
    for j in range(m):
        A[i][j] = B[j]

for i in range(n):
    for j in range(m):
        A[i][m] += A[i][j]

for j in range(m+1):
    for i in range(n):
        A[n][j] += A[i][j]

for i in range(n+1):
    print(A[i][0], end='')
    for j in range(1, m):
        print(' %d'%A[i][j], end='')
    print()

"""
Input:
4 5
1 1 3 4 5
2 2 2 4 5
3 3 0 1 1
2 3 4 4 6

Output:
1 1 3 4 5
2 2 2 4 5
3 3 0 1 1
2 3 4 4 6
8 9 9 13 17
"""

どこがミスっているのか？　とざっと見て、出力時のループでrange(1, m)というのを見つけたのでrange(1, m+1)にします。

for i in range(n+1):
    print(A[i][0], end='')
    for j in range(1, m+1):
        print(' %d'%A[i][j], end='')
    print()

"""
Input:
4 5
1 1 3 4 5
2 2 2 4 5
3 3 0 1 1
2 3 4 4 6

Output:
1 1 3 4 5 14
2 2 2 4 5 15
3 3 0 1 1 8
2 3 4 4 6 19
8 9 9 13 17 56
"""

できました。これでよしとするなら良いですし、まあ私なら読みづらいのでリファクタリングします。

Comments

[chibiyt]

回答ありがとうございました！
ものすごく初歩的なところでミスしていてお恥ずかしい限りです…

もしよろしければリファクタリングの案なども教えて頂けますと幸いです。

[hayataka2049]

とりあえず入力は先に済ませた方がわかりやすいんじゃないでしょうか。
n, m = map(int, input().split())
A = [list(map(int, input().split()))
for _ in range(n)]
みたいな感じに。
行方向はsumでそのままできます。列方向はzipを使ってやるいい方法があります（「python list 転置」などで検索すると出るでしょう）。

[chibiyt]

なるほど。
確かに合計出すだけならsum使えば済む話でしたね。
とても勉強になりました！
丁寧にリファクタリングまで教えてくださりありがとうございました！！