读取txt文件输入并将值添加到两个数组

Question

A   B   C  D

2   4   5  6

4   5   3  7

3   6   7  8

我要将A、B、C列值获取到数组（3 x 3），将D列获取到另一个数组（3 x 1）。你知道吗

Answer 1

简单暴力法：

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']

Answer 2

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)