1. 程式人生 > >Python3 中把txt資料檔案讀入到矩陣中

Python3 中把txt資料檔案讀入到矩陣中

1.例項程式:

'''
資料檔案:2.txt內容:(以空格分開每個資料)
1 2 2.5
3 4 4
7 8 7
'''

from numpy import *
A = zeros((3,3),dtype=float)    #先建立一個 3x3的全零方陣A,並且資料的型別設定為float浮點型

f = open('2.txt')               #開啟資料檔案檔案
lines = f.readlines()           #把全部資料檔案讀到一個列表lines中
A_row = 0                       #表示矩陣的行,從0行開始
for line in lines:              #把lines中的資料逐行讀取出來
    list = line.strip('\n').split(' ')      #處理逐行資料:strip表示把頭尾的'\n'去掉,split表示以空格來分割行資料,然後把處理後的行資料返回到list列表中
    A[A_row:] = list[0:3]                    #把處理後的資料放到方陣A中。list[0:3]表示列表的0,1,2列資料放到矩陣A中的A_row行
    A_row+=1                                #然後方陣A的下一行接著讀
    #print(line)

print(A)    #列印 方陣A裡的資料

列印結果:
[[ 1.   2.   2.5]
 [ 3.   4.   4. ]
 [ 7.   8.   7. ]]
2.資料讀入矩陣的邏輯:

為簡單解釋,例如我們要把:

1 2 3

4 5 6

7 8 9

讀入矩陣中,以上述的程式碼為例: 

當 A_row =0時,執行 A[A_row:] = list[0:3後矩陣A是:

1 2 3
1 2 3
1 2 3

當A_row = 1是,執行 A[A_row:] = list[0:3] 後矩陣A是:

1 2 3
4 5 6
4 5 6

當A_row = 2時,執行 A[A_row:] = list[0:3] 後矩陣A是:

1 2 3
4 5 6
7 8 9

也就是上述程式碼:

for line in lines:                      #先把逐行資料取出來
    list = line.strip('\n').split(' ')      #再通過處理,放回到list列表中
    A[A_row:] = list[0:3]                    #然後把list列表的資料放到矩陣中
    A_row+=1                
所做的事