Python - 将 CSV 列读入列表

在数据处理和分析中，常常需要从CSV（逗号分隔值）文件中读取数据并将其转换为Python中的列表。在Python中，有很多库可以用来读取CSV文件，但是最常用的是内置的csv库。

使用内置的csv库

import csv
    
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上述代码打开名为"data.csv"的文件并读取其中的所有行。由于内置的csv.reader函数返回一个迭代器，我们可以使用for循环迭代每一行并将其打印出来。

如果数据文件含有一行标题，我们可以使用csv库的DictReader函数将其转换为Python字典。这样可以更方便地访问行数据。

import csv
    
with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)

上述代码将"data.csv"文件读取为字典，其中字典的键为数据文件中的标题。

将CSV列读入列表

有时我们只需要CSV文件中的一部分数据，比如某一列。我们可以使用以下代码将CSV中的某一列读取到Python列表中。

import csv
    
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    column = [row[0] for row in reader]
    print(column)

上述代码将"data.csv"文件的第一列读取为Python列表。

总结

Python内置的csv模块可以帮助我们轻松地读取CSV文件中的数据并将其转换为Python中的列表或字典，非常实用。