数学 ガウスの消去法 txtファイル読み込み 行列

Question

前提・実現したいこと

ガウスの消去法で三元連立方程式を解くプログラムを組んでいて、各式の値をテキストファイルから読み込みたい。（下のソースコードにおける、行列a,bを、ソースコード中で値を指定するのではなく、予め各値を書き込んだテキストファイルの読み込みにより作りたいです）

該当のソースコード

import numpy as np

def g(a,b):
    a_array = np.array(a)
    b_array = np.array(b)

    for i in range(1,3):
        for j in range(i,3):
            k = a_array[j,i-1]/a_array[i-1,i-1]
            a_array[j] = a_array[j] - a_array[i-1]*k
            b_array[j] = b_array[j] - b_array[i-1]*k

    for i in range(1,-1,-1):
        for j in range(i,-1,-1):
            k = a_array[j,i+1] / a_array[i+1,i+1]
            a_array[j] = a_array[j] - a_array[i+1]*k
            b_array[j] = b_array[j] - b_array[i+1]*k

    x=[]
    for i in range(3):
        x.append(round((b_array[i]/a_array[i,i]),6))
    return x

a = [[2.0,3.0,-1.0], [5.0,-1.0,2.0], [-1.0,1.0,5.0]]
b = [21.0,1.0,-7.0]
with open(' ⓶解.txt', 'w') as t:
    print(g(a,b),file=t)

試したこと

ファイル読み込みのためにreadlineなど色々試しましたが、良い方法が思いつきませんでした。何か良い方法がありましたらご教授お願いします。

Answer 1

numpyを使っているのであればnumpy.loadtxtを使うのが簡便でよいと思います。
読み込みたいファイル形式もある程度引数で自由に指定できます。

Comments

[Etori]

試したらうまく行きました。ありがとうございました。