Question

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

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

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

Answer 1

您可以使用用C编写的pandas包，以便快速使用read_csv和usecols来指定您需要的列：

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

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