Python获取CSV一列数据|极客教程

Python获取CSV一列数据

在数据处理过程中，经常需要从CSV文件中获取特定列的数据，Python提供了方便的工具来实现这个目的。本文将介绍如何使用Python来获取CSV文件中的一列数据。

读取CSV文件

首先，我们需要使用Python的标准库csv来读取CSV文件。在读取CSV文件之前，我们需要先导入csv模块，并使用csv.reader函数来读取文件。

import csv

# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

运行结果：

['Name', 'Age', 'Gender']
['Alice', '25', 'Female']
['Bob', '30', 'Male']
['Cathy', '28', 'Female']

以上示例代码读取了一个CSV文件，并打印了文件中的所有行数据。接下来，我们将介绍如何获取特定列的数据。

获取特定列数据

如果我们需要获取CSV文件中特定一列的数据，我们可以使用以下方法：

import csv

# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)

    # 获取特定列数据
    column_data = [row[0] for row in reader]
    print(column_data)

运行结果：

['Name', 'Alice', 'Bob', 'Cathy']

以上示例代码通过列表推导式的方式，将CSV文件中第一列的数据存储在column_data列表中。接下来，我们将介绍如何处理包含geek-docs.com的字符串。

处理包含指定字符串的数据

有时候，我们需要筛选出包含指定字符串的数据。我们可以使用以下示例代码来实现这个目的：

import csv

# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)

    # 筛选包含指定字符串的数据
    target_string = 'geek-docs.com'
    filtered_data = [row[0] for row in reader if target_string in row[0]]
    print(filtered_data)

运行结果：