Enthought Canopy中的Python:操纵csv文件

时间:2015-12-07 05:41:10

标签: python csv

我的代码目标是能够引入一个csv文件,只获取包含坐标值的列,将这些列转换为数组,并使用另一个文件中的另一组坐标执行数学函数。 python中的文件操作不是我的强项。这是我的代码:

    list1 = []
    DR12 = open('dr12data.csv', 'r')
    for line in DR12:
        linelist = line.split(',')
        list1.append(linelist)

这会将文件中的每一行转换为二维列表,在该列表中我可以选择我想要的各个值。但是,它不允许我一次选择多个值,如果我想将我的坐标放在一个单独的文件或列表中,我将不得不逐行(一个更强力的方法)。我正在努力使我的代码更有效率和更通用。

1 个答案:

答案 0 :(得分:0)

您可以使用用C编写的pandas包,以便快速使用read_csvusecols来指定您需要的列:

import pandas as pd
df = pd.read_csv('your_file', usecols=['col1', 'col2'])

然后你可以使用很多的pantmethods,用整列做数学函数。