Enthought中的Python :操作csv文件

Question

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

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

这会将文件中的每一行转换为一个二维列表，我可以从中挑选出我想要的各个值。但是，它不允许我一次选择多个值，如果我想将坐标放在一个单独的文件或列表中，我必须逐行选择(一种更暴力的方法)。我正在努力让我的代码更高效、更通用。

Answer 1

您可以使用pandas包，它是用C编写的，所以它很快，并结合使用read_csv和usecols来指定您需要的列：

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

然后你可以使用很多pantmethod，对整个列做数学函数。