Python CSV 查找特定字符串

CSV是一种常见的文件格式，往往用于数据存储与交换。在Python中，我们可以使用csv模块来读取和写入CSV文件。

在许多情况下，我们需要对CSV文件进行搜索，以找到特定的字符串或者进行数据过滤。下面我们将介绍如何使用Python来查找CSV文件中的特定字符串。

读取CSV文件

首先，我们需要使用csv模块来读取CSV文件，并将其转换为Python中的列表。

import csv

with open('example.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    data = [row for row in csv_reader]

在上面的示例中，我们打开了一个名为example.csv的CSV文件。csv.reader函数将CSV文件转换为迭代器。我们可以通过使用列表推导式将其转换为Python中的列表。

查找特定字符串

有了Python中的列表表示CSV文件，我们可以使用通常的字符串搜索方法来查找特定字符串。例如，下面的代码将返回包含指定子字符串的所有行：

data = [row for row in data if 'example' in row]

在上面的示例中，我们将列表中的每一行与指定的子字符串进行比较。如果包含该子字符串，就将该行添加到新的列表中。

写入CSV文件

最后，我们可以将筛选后的数据写入一个新的CSV文件。下面的示例演示了如何创建一个新的CSV文件，并将筛选后的数据写入其中：

import csv

with open('new_file.csv', 'w', newline='') as csv_file:
    csv_writer = csv.writer(csv_file)
    csv_writer.writerows(data)

在上面的示例中，csv.writer函数将Python中的列表转换为CSV文件的形式，并将其写入到新的CSV文件中。

总结：

通过以上的介绍，我们可以看出使用Python来查找CSV文件中的特定字符串非常简单。我们只需要将CSV文件转换为Python中的列表，然后可以使用任何Python字符串方法来查找并筛选特定的数据。最后，我们可以将筛选后的数据写入一个新的CSV文件中。

代码片段

import csv

# 读取CSV文件
with open('example.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    data = [row for row in csv_reader]

# 查找包含指定子字符串的行
data = [row for row in data if 'example' in row]

# 将筛选后的数据写入新的CSV文件
with open('new_file.csv', 'w', newline='') as csv_file:
    csv_writer = csv.writer(csv_file)
    csv_writer.writerows(data)