尝试在python中追加数据文件中包含多行的列表

时间:2014-03-06 22:26:25

标签: python

所以这可能有点难以描述

假设我有1000行的csv fie。和5列。我想做的就是拿走 第5列中的每个值,并将其附加到列表中。但是,我想将该列表中的所有值分开5。像

这样的东西
[[1,1,1,1,1][2,2,2,2,2][3,3,3,3,3][4,4,4,4,4]  . . . . .]

1 个答案:

答案 0 :(得分:0)

你可能一下子就可以做到这一点,但除非速度至关重要,否则我认为两次传球更加清晰。首先,将所需数据读入平面列表,然后将该列表分为五个项目的子列表。

with open(csvfile, 'r') as f:
    reader = csv.reader(f)

    # grab fifth column from each row
    data = [row[4] for row in reader] 

    # Split the flat data into sublists of length five
    split_data = [[data[i*j] for i in range(5)] for j in range(len(data)//5)]