面试题：Python中如何使用csv模块读取CSV文件特定列的数据

Question

假设存在一个CSV文件，每行数据包含姓名、年龄、性别等多列信息。请使用Python的csv模块编写代码，读取该CSV文件，并仅提取年龄这一列的数据，最后将其以列表形式输出。

Answer 1

import csv


def get_ages():
    ages = []
    with open('your_file.csv', 'r', encoding='utf-8') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            ages.append(row['年龄'])
    return ages

你可以使用以下方式调用函数：

if __name__ == '__main__':
    print(get_ages())

在上述代码中：

1. 首先导入csv模块。
2. get_ages函数中，使用with语句打开CSV文件，并使用csv.DictReader以字典形式读取每一行数据，这样可以通过列名来访问对应的数据。
3. 遍历每一行数据，将其中的年龄列数据添加到ages列表中。
4. 最后返回包含所有年龄数据的列表。
5. 在if __name__ == '__main__':块中调用get_ages函数并打印结果。请将'your_file.csv'替换为真实的CSV文件名。