欢迎访问宙启技术站
智能推送

利用load_data函数读取CSV文件的Python代码实现

发布时间:2023-12-13 17:51:26

以下是使用Python编写的load_data函数,用于读取CSV文件的内容:

import csv

def load_data(filename):
    data = []
    with open(filename, 'r', newline='') as file:
        reader = csv.reader(file)
        for row in reader:
            data.append(row)
    return data

在这个函数中,我们首先通过open函数打开指定的CSV文件,并指定以只读模式打开。在csv.reader函数中传入打开的文件对象,这样我们就创建了一个CSV reader对象来读取文件内容。然后我们使用for循环遍历每一行,并将其添加到一个名为data的列表中。

最后,我们将读取的数据列表返回。

下面是load_data函数的使用示例:

data = load_data('data.csv')
for row in data:
    print(row)

上面的代码将会打开名为"data.csv"的文件,并将其内容读取到一个名为data的列表中。然后,我们使用for循环遍历data列表打印每一行的内容。

假设我们的data.csv文件内容如下:

1,John,Doe
2,Jane,Smith
3,David,Johnson

运行上述代码的输出结果将会是:

['1', 'John', 'Doe']
['2', 'Jane', 'Smith']
['3', 'David', 'Johnson']

这些输出行对应于文件中的每一行数据。请注意,数据项都是字符串类型,而不是整数或其他类型。如果你需要将这些数据项转换为其他类型,比如整数,你可以在读取数据的时候进行类型转换。

只需将相应的数据项转换为整数即可,例如:

for row in data:
    row[0] = int(row[0])

这样,我们就把第一列的数据从字符串转换为整数类型。

以上就是使用Python实现的load_data函数,并提供了一个简单的使用例子。