读取txt文件输入和添加值以两个阵列

Question

A B C D 

2 4 5 6 

4 5 3 7 

3 6 7 8 

我想获得A，B，C列值阵列（3×3）和D列到另一阵列（3×1）。

Answer 1

import numpy as np  

# Read the data from a file 
with open('data.txt') as file: 
    lines = file.readlines() 

# Chop of the columns 
raw_data = lines[1:] 

# Now fetch all the data 
data_abc = [] 
data_d = [] 

for line in raw_data: 
    values = line.split() 
    data_abc.append(values[:3]) 
    data_d.append(values[3]) 

# Convert to matrix 
data_abc = np.asmatrix(data_abc) 
data_d = np.asmatrix(data_d) 

# Display the result 
print('Data A B C:', data_abc) 
print('Data D:', data_d) 

Answer 2

简单穷举法：

a33 = [[],[],[]] 
a31 = [] 

with open('dat.txt') as f: 
    for ln in f: 
     a,b,c,d = ln.split() 
     a33[0] += a 
     a33[1] += b 
     a33[2] += c 
     a31 += d 

print a33 
print a31 

[['2', '4', '3'], ['4', '5', '6'], ['5', '3', '7']] 
['6', '7', '8']