csv - 定义一个打开 csv 文件并返回文件的函数

Question

我一直在思考如何处理这段代码。

def getFile(filename):
    with open (filename, 'r') as reader:
         csv_reader = csv.reader(reader)
         header = next(csv_reader)
    return csv_reader

file = getFile(filename)
list = []
def getpointlist(file):
    
    for line in file:
        name = line[0]
        age = float(line[1])
        number = float(line[2])
        tup = (name, age, number)
        list.append(tup)

我想要的是使用一个函数打开一个文件，然后将它传递给另一个函数。我总是在关闭文件错误时获得 I/O 操作。泰！

标签： csv

Answer 1

你试过用熊猫吗？

import pandas as pd

file= pd.read_csv('test.csv', delimiter=';')

print(file.head())

结果：

   name   age test
0  jojo  34.0   ER
1  jojo  34.0   ER
2  jojo  34.0   ER
3  jojo  34.0   ER
4  jojo  34.0   ER