Python 使用 csv 导入数据到列表
在本文中,我们将介绍如何使用 Python 的内置库 csv 将数据从 csv 文件中导入并存储到列表中。csv 是一种常见的文件格式,用于存储表格数据。通过将数据导入列表,我们可以方便地对数据进行处理和分析。
阅读更多:Python 教程
csv 文件的基本结构
csv 文件以逗号分隔各个字段,并以换行符分隔各个行。每一行都表示一条数据记录,每个字段对应了该行中的一个数值。下面是一个示例 csv 文件的结构:
name,age,gender
John,25,Male
Alice,30,Female
Bob,40,Male
在上面的示例中,第一行是表头,包含了三个字段:name, age 和 gender。接下来的每一行都包含了具体的数据记录。
使用 csv 库读取 csv 文件
首先,我们需要导入 csv 库。Python 的 csv 库提供了用于读取和写入 csv 文件的功能。
import csv
要读取 csv 文件,我们可以使用 reader
函数。该函数接受一个文件对象作为参数,并返回一个可以迭代的 csv 对象。我们可以通过循环遍历 csv 对象来访问每一行的数据。
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
print(row)
上面的代码将打开名为 data.csv 的文件,并逐行打印文件中的数据。在每一次迭代中,row
变量将包含当前行的数据作为一个列表。
将 csv 数据导入列表
为了将 csv 数据导入列表,我们可以创建一个空列表,并在循环中将每一行的数据附加到列表中。以下是一个示例代码:
data_list = []
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
data_list.append(row)
print(data_list)
上面的代码将创建一个名为 data_list
的列表,并将每一行的数据从 csv 文件中添加到该列表中。最后,代码将打印整个列表。
示例:计算平均年龄
让我们通过一个示例来说明如何使用导入的列表进行数据处理。我们将计算上面示例 csv 文件中所有人的平均年龄。
ages = []
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
next(csv_reader) # 跳过第一行表头
for row in csv_reader:
ages.append(int(row[1])) # 添加第二列(年龄)的数据到 ages 列表
average_age = sum(ages) / len(ages)
print("平均年龄:", average_age)
在上面的代码中,我们创建了一个名为 ages
的空列表,然后在循环中将每一行的第二列(即年龄)添加到列表中。最后,我们计算了列表 ages
中所有元素的平均值,并将结果打印出来。
总结
本文介绍了如何使用 Python 的 csv 库将数据从 csv 文件导入并存储到列表中。通过将数据导入列表,我们可以方便地对数据进行处理和分析。我们学习了读取 csv 文件的基本结构,使用 csv 库读取 csv 文件的方法,以及将 csv 数据导入列表的步骤。最后,我们通过一个示例演示了如何使用导入的列表进行数据处理。希望本文对你理解和使用 Python 中的 csv 导入功能有所帮助。