python按列读取文本数据

工作需要按照列读取文本文件，解决过程记录如下：

（1）易读写法：

import csv
import numpy as np
with open('imu0.csv','r') as f:
    timestamps=[]
    omega=[]
    alpha=[]
    for row in f.readlines():
        row=row.split(',')
        timestamps.append(row[0])
        omega.append(row[1:4])
        alpha.append(row[4:7])
    timestamps = np.array(map(eval, timestamps))
    #omega = np.array(map(eval, omega))
print(timestamps)
print(omega)

这里按照行读取数据，然后按照逗号分割每行的数据，添加到对应列表中。

但是读取的数据是str类型，需要转为int。这里使用：

map(eval, timestamps)

但是omega无法使用map这种方法，因为其每个元素又是list。

因此可以改为：

import csv
import numpy as np
with open('imu0.csv','r') as f:
    timestamps=[]
    omega=[]
    alpha=[]
    for row in f.readlines():
        row=row.split(',')
        timestamps.append(int(row[0]))
        omega.append(map(eval,row[1:4]))
        alpha.append(map(eval,row[4:7]))