📜  如何在Python读取 CSV 文件中的数字?(1)

📅  最后修改于: 2023-12-03 15:24:36.188000             🧑  作者: Mango

如何在Python读取 CSV 文件中的数字?

在Python中,读取CSV文件中的数字是一个非常普遍的任务。以下是一些方法:

使用csv库

Python的csv模块可以帮助您轻松地读取CSV文件中的数字。您可以使用该模块的csv.reader()方法,将CSV文件读入列表并将其作为数字列表返回。以下是一个例子:

import csv

with open('file.csv', 'r') as f:
    csv_reader = csv.reader(f)
    data = []
    for row in csv_reader:
        row_data = []
        for item in row:
            row_data.append(int(item))
        data.append(row_data)

在这个例子中,我们首先将CSV文件打开并使用csv.reader()方法创建一个csv_reader对象。然后我们遍历csv_reader对象,并在每一行的每个项目上运行int()函数以将其转换为一个整数。最后,我们将每一行作为数字列表添加到data列表中。

使用pandas库

另一个常见的Python库是pandas,它可以帮助您更轻松地读取和处理CSV文件中的数字。与csv模块不同,pandas库可以将CSV文件读入为DataFrame对象。

import pandas as pd

data = pd.read_csv('file.csv', sep=',', header=None, dtype=int)

在这个例子中,我们使用pd.read_csv()方法从文件中读取CSV数据,并将结果存储在data变量中。在这个方法中,我们使用sep参数来指定CSV文件的分隔符,header=None参数来指示文件没有标题行,dtype=int参数来指示数据应该被读取为整数类型。

总结

无论您选择使用csv模块还是pandas库,读取CSV文件中的数字都是一个非常简单和常见的任务。这些库都提供了简单易用的方法来读取数值数据,并将它们存储在方便处理的数据结构中。